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
 
Node * return_node ()
 
const Node * return_node () const
 
Node * param_node ()
 
const Node * param_node () const
 
Graph * owningGraph ()
 
const Graph * owningGraph () const
 
Node * owningNode ()
 
const Node * owningNode () const
 
Value * addInput (std::string name="")
 
Value * insertInput (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)
 
Node * appendNode (Node *n)
 
Node * prependNode (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: