Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Friends
torch::jit::Block Struct Reference

Public Member Functions

 TH_DISALLOW_COPY_AND_ASSIGN (Block)
 
TORCH_API Block (Graph *graph_, Node *node_)
 
at::ArrayRef< Value * > inputs ()
 
at::ArrayRef< const Value * > inputs () const
 
at::ArrayRef< Value * > outputs ()
 
at::ArrayRef< const Value * > outputs () const
 
graph_node_list nodes ()
 
const_graph_node_list nodes () const
 
Nodereturn_node ()
 
const Nodereturn_node () const
 
Nodeparam_node ()
 
const Nodeparam_node () const
 
GraphowningGraph ()
 
const GraphowningGraph () const
 
NodeowningNode ()
 
const NodeowningNode () const
 
ValueaddInput (std::string name="")
 
ValueinsertInput (size_t i, std::string name="")
 
void eraseInput (size_t i)
 
size_t registerOutput (Value *v)
 
size_t insertOutput (size_t i, Value *n)
 
void eraseOutput (size_t i)
 
NodeappendNode (Node *n)
 
NodeprependNode (Node *n)
 
TORCH_API void cloneFrom (Block *src, std::function< Value *(Value *)> value_map)
 

Friends

struct Node
 
struct Graph
 

Detailed Description

Definition at line 818 of file ir.h.


The documentation for this struct was generated from the following files: