 |
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
|
Caffe2 - C++ API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| TH_DISALLOW_COPY_AND_ASSIGN (Node) | |
| Node *& | next () |
| Node *& | prev () |
| Node *const & | next () const |
| Node *const & | prev () const |
| NodeKind | kind () const |
| Node * | setSourceLocation (std::shared_ptr< SourceLocation > sl) |
| std::shared_ptr< SourceLocation > | getSourceLocation () const |
| Graph * | owningGraph () |
| const Graph * | owningGraph () const |
| Block * | owningBlock () |
| const Block * | owningBlock () const |
| ScopePtr | scope () |
| void | setScope (ScopePtr scope) |
| std::string | scopeName () const |
| at::ArrayRef< Value * > | inputs () |
| at::ArrayRef< const Value * > | inputs () const |
| at::ArrayRef< Value * > | outputs () |
| at::ArrayRef< const Value * > | outputs () const |
| Value * | output (size_t i) const |
| bool | hasUses () const |
| TORCH_API void | replaceAllUsesWith (Node *n) |
| Value * | input () |
| Value * | output () |
| const Value * | output () const |
| const Value * | input () const |
| Value * | input (size_t i) const |
| Value * | namedInput (Symbol name) const |
| c10::optional< IValue > | get (Symbol name) const |
| template<typename T > | |
| c10::optional< T > | get (Symbol name) const |
| bool | is_constant (Symbol name) const |
| TORCH_API bool | mustBeNone () const |
| TORCH_API bool | isNondeterministic () const |
| TORCH_API bool | hasSideEffects () const |
| TORCH_API Value * | addInput (Value *value) |
| TORCH_API Value * | insertInput (size_t i, Value *value) |
| TORCH_API Value * | replaceInput (size_t i, Value *newValue) |
| TORCH_API void | replaceInputWith (Value *from, Value *to) |
| TORCH_API Value * | addOutput () |
| TORCH_API Value * | insertOutput (size_t i) |
| TORCH_API void | eraseOutput (size_t i) |
| TORCH_API Block * | addBlock () |
| TORCH_API void | eraseBlock (size_t i) |
| at::ArrayRef< Block * > | blocks () |
| at::ArrayRef< const Block * > | blocks () const |
| TORCH_API bool | isBefore (const Node *n) const |
| TORCH_API bool | isAfter (const Node *n) const |
| TORCH_API Node * | insertBefore (Node *n) |
| TORCH_API Node * | insertAfter (Node *n) |
| TORCH_API void | moveAfter (Node *n) |
| TORCH_API void | moveBefore (Node *n) |
| TORCH_API void | removeInput (size_t i) |
| TORCH_API void | removeAllInputs () |
| graph_node_list_iterator | iterator () |
| graph_node_list_iterator | reverseIterator () |
| const_graph_node_list_iterator | iterator () const |
| const_graph_node_list_iterator | reverseIterator () const |
| TORCH_API void | destroy () |
| template<typename T > | |
| T * | cast () |
| template<typename T > | |
| T * | expect () |
| TORCH_API bool | matches (const char *signature_literal, at::ArrayRef< Symbol > const_inputs={}) const |
| const FunctionSchema & | schema () const |
| const FunctionSchema * | maybeSchema () const |
| void | dump () const |
| std::ostream & | print (std::ostream &out, size_t level, std::vector< const Node * > *groups) const |
| void | copyAttributes (const Node &rhs) |
| bool | hasAttribute (Symbol name) const |
| bool | hasAttributeS (const std::string &name) const |
| AttributeKind | kindOf (Symbol name) const |
| AttributeKind | kindOfS (const std::string &name) const |
| Node * | removeAttribute (Symbol name) |
| Node * | removeAttributeS (const std::string &name) |
| bool | hasAttributes () const |
| size_t | numAttributes () const |
| std::vector< Symbol > | attributeNames () const |
| std::vector< const char * > | attributeNamesS () const |
| GraphAttr::ValueType & | g (Symbol name) |
| Node * | t_ (Symbol name, TensorAttr::ConstructorType v) |
| const TensorAttr::ValueType & | t (Symbol name) const |
| Node * | ts_ (Symbol name, TensorsAttr::ConstructorType v) |
| const TensorsAttr::ValueType & | ts (Symbol name) const |
Data Fields | |
| friend | graph_node_list |
| friend | const_graph_node_list |
| friend | graph_node_list_iterator |
| friend | const_graph_node_list_iterator |
| Node * | next_in_graph [2] = {nullptr, nullptr} |
Protected Member Functions | |
| TORCH_API | Node (Graph *graph_, NodeKind kind_) |
| virtual Node * | allocNewInstance (Graph *g) |
| virtual TORCH_API void | cloneFrom (Node *s) |
Friends | |
| struct | Graph |
| struct | Block |
| struct | Value |
1.8.11
