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 |