|
const NodeTypeInfo & | get_type_info () const override |
|
| ScatterNDUpdate (const Output< Node > &inputs, const Output< Node > &indices, const Output< Node > &updates) |
|
virtual std::shared_ptr< Node > | clone_with_new_inputs (const OutputVector &new_args) const override |
|
const NodeTypeInfo & | get_type_info () const override |
|
virtual void | validate_and_infer_types () override |
| Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times. More...
|
|
virtual bool | visit_attributes (AttributeVisitor &visitor) override |
|
void | constructor_validate_and_infer_types () |
|
virtual const op::AutoBroadcastSpec & | get_autob () const |
|
virtual bool | evaluate (const HostTensorVector &output_values, const HostTensorVector &input_values) const |
| Evaluates the op on input_values putting results in output_values. More...
|
|
virtual bool | evaluate_lower (const HostTensorVector &output_values) const |
|
virtual bool | evaluate_upper (const HostTensorVector &output_values) const |
|
virtual bool | constant_fold (OutputVector &output_values, const OutputVector &inputs_values) |
|
virtual OutputVector | decompose_op () const |
| Decomposes the FusedOp into a sub-graph consisting of core ngraph ops. More...
|
|
const char * | get_type_name () const |
|
void | set_arguments (const NodeVector &arguments) |
| Sets/replaces the arguments with new arguments.
|
|
void | set_arguments (const OutputVector &arguments) |
| Sets/replaces the arguments with new arguments.
|
|
void | set_argument (size_t position, const Output< Node > &argument) |
| Sets/replaces the arguments with new arguments.
|
|
void | set_output_type (size_t i, const element::Type &element_type, const PartialShape &pshape) |
|
void | set_output_size (size_t output_size) |
| Sets the number of outputs.
|
|
void | invalidate_values () |
|
virtual void | revalidate_and_infer_types () |
|
virtual std::string | description () const |
| Get the string name for the type of the node, such as Add or Multiply . The class name, must not contain spaces as it is used for codegen. More...
|
|
const std::string & | get_name () const |
| Get the unique name of the node. More...
|
|
void | set_friendly_name (const std::string &name) |
| Sets a friendly name for a node. This does not overwrite the unique name of the node and is retrieved via get_friendly_name(). Used mainly for debugging. The friendly name may be set exactly once. More...
|
|
const std::string & | get_friendly_name () const |
| Gets the friendly name for a node. If no friendly name has been set via set_friendly_name then the node's unique name is returned. More...
|
|
virtual bool | is_dynamic () const |
|
size_t | get_instance_id () const |
|
virtual std::ostream & | write_description (std::ostream &os, uint32_t depth=0) const |
| Writes a description of a node to a stream. More...
|
|
const std::vector< std::shared_ptr< Node > > & | get_control_dependencies () const |
| Get control dependencies registered on the node.
|
|
const std::vector< Node * > & | get_control_dependents () const |
| Get nodes dependent on this node.
|
|
void | add_control_dependency (std::shared_ptr< Node > node) |
| This node cannot execute until node executes.
|
|
void | remove_control_dependency (std::shared_ptr< Node > node) |
| Remove the dependency of this node on node.
|
|
void | clear_control_dependencies () |
| Remove all dependencies from this node.
|
|
void | clear_control_dependents () |
| Remove this node as a dependency from all dependent nodes.
|
|
void | add_node_control_dependencies (std::shared_ptr< Node > source_node) |
| This node absorbs the control dependencies of source_node.
|
|
void | add_node_control_dependents (std::shared_ptr< Node > source_node) |
| This node becomes a dependent of every node dependent on source_node.
|
|
void | transfer_control_dependents (std::shared_ptr< Node > replacement) |
| This node's control dependencies are replaced by replacement.
|
|
size_t | get_output_size () const |
| Returns the number of outputs from the node.
|
|
const element::Type & | get_output_element_type (size_t i) const |
| Returns the element type for output i.
|
|
const element::Type & | get_element_type () const |
| Checks that there is exactly one output and returns its element type.
|
|
const Shape & | get_output_shape (size_t i) const |
| Returns the shape for output i.
|
|
const PartialShape & | get_output_partial_shape (size_t i) const |
| Returns the partial shape for output i.
|
|
Output< const Node > | get_default_output () const |
|
Output< Node > | get_default_output () |
|
virtual size_t | get_default_output_index () const |
| Returns the output of the default output, or throws if there is none.
|
|
size_t | no_default_index () const |
| Throws no default.
|
|
const Shape & | get_shape () const |
| Checks that there is exactly one output and returns its shape.
|
|
descriptor::Tensor & | get_output_tensor (size_t i) const |
| Returns the tensor for output or input i.
|
|
descriptor::Tensor & | get_input_tensor (size_t i) const |
|
size_t | get_input_size () const |
| Returns the number of inputs for the op.
|
|
const element::Type & | get_input_element_type (size_t i) const |
| Returns the element type of input i.
|
|
const Shape & | get_input_shape (size_t i) const |
| Returns the shape of input i.
|
|
const PartialShape & | get_input_partial_shape (size_t i) const |
| Returns the partial shape of input i.
|
|
Node * | get_input_node_ptr (size_t index) const |
|
std::shared_ptr< Node > | get_input_node_shared_ptr (size_t index) const |
|
Output< Node > | get_input_source_output (size_t i) const |
|
std::shared_ptr< Node > | copy_with_new_inputs (const OutputVector &new_args) const |
|
std::shared_ptr< Node > | copy_with_new_inputs (const OutputVector &inputs, const std::vector< std::shared_ptr< Node >> &control_dependencies) const |
|
bool | has_same_type (std::shared_ptr< const Node > node) const |
| True if this and node have one output with same element type and shape.
|
|
RTMap & | get_rt_info () |
|
const RTMap & | get_rt_info () const |
|
const std::unordered_set< std::string > & | get_provenance_tags () const |
|
void | add_provenance_tag (const std::string &tag) |
|
template<typename T > |
void | add_provenance_tags (T tag_set) |
|
void | add_provenance_tags_above (const OutputVector &base, const std::unordered_set< std::string > &tag_set) |
| Adds tag_set to this node and all intermediate nodes above base.
|
|
void | remove_provenance_tag (const std::string &tag) |
|
void | add_provenance_group_member (const std::shared_ptr< Node > &node) |
| Add node to additional nodes that receive tags.
|
|
void | remove_provenance_group_member (const std::shared_ptr< Node > &node) |
| Remove node to additional nodes that receive tags.
|
|
void | replace_provenance_group_member (const std::shared_ptr< Node > ¤t_node, const std::shared_ptr< Node > &replacement_node) |
| Replace current_node with replacement_node and transfer tags.
|
|
const std::set< std::shared_ptr< Node > > & | get_provenance_group_members () const |
|
std::shared_ptr< Node > | add_provenance_group_members_above (const OutputVector &base) |
| Add all nodes between this node and nodes in base as additional nodes to receive provenance tags.
|
|
void | merge_provenance_tags_from (const std::shared_ptr< const Node > &source) |
|
void | transfer_provenance_tags (const std::shared_ptr< Node > &replacement) |
| Transfer provenance tags to replacement.
|
|
NodeVector | get_users (bool check_is_used=false) const |
| Get all the nodes that uses the current node.
|
|
virtual size_t | get_version () const |
|
virtual std::shared_ptr< Node > | get_default_value () const |
|
bool | operator< (const Node &other) const |
| Use instance ids for comparison instead of memory addresses to improve determinism.
|
|
std::vector< Input< Node > > | inputs () |
|
std::vector< Input< const Node > > | inputs () const |
|
std::vector< Output< Node > > | input_values () const |
|
std::vector< Output< Node > > | outputs () |
|
std::vector< Output< const Node > > | outputs () const |
|
Input< Node > | input (size_t input_index) |
|
Input< const Node > | input (size_t input_index) const |
|
Output< Node > | input_value (size_t input_index) const |
|
Output< Node > | output (size_t output_index) |
|
Output< const Node > | output (size_t output_index) const |
|
void | set_op_annotations (std::shared_ptr< ngraph::op::util::OpAnnotations > op_annotations) |
|
std::shared_ptr< ngraph::op::util::OpAnnotations > | get_op_annotations () const |
|
virtual bool | match_value (pattern::Matcher *matcher, const Output< Node > &pattern_value, const Output< Node > &graph_value) |
|
virtual bool | match_node (pattern::Matcher *matcher, const Output< Node > &graph_value) |
|