|
|
| PythonOp (Graph *graph) |
| |
|
PythonOp * | init (THPObjectPtr &&pyobj, const std::string &cconv, pyobj_list &&scalar_args) |
| |
|
virtual std::string | name () const =0 |
| |
|
virtual void | writeScalars (std::ostream &out) const =0 |
| |
|
void | cloneFrom (Node *other_) override=0 |
| |
|
Node * | allocNewInstance (Graph *g) override=0 |
| |
|
virtual c10::optional< THPObjectPtr > | autogradFunction () const =0 |
| |
|
| 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 |
| |
Definition at line 1236 of file ir.h.
Public Member Functions inherited from torch::jit::Node
1.8.11
