Public Member Functions | Protected Attributes
ngraph::descriptor::Output Class Reference

Public Member Functions

 Output (Node *node, size_t index, const std::shared_ptr< Tensor > &tensor)
 
std::shared_ptr< Node > get_node () const
 
size_t get_index () const
 
ngraph::Output< Node > get_output () const
 
std::shared_ptr< Tensorget_tensor_ptr () const
 
void set_tensor_ptr (const std::shared_ptr< Tensor > &tensor)
 
void add_input (Input *input)
 
void remove_input (Input *input)
 
const std::vector< Input * > & get_inputs () const
 
Tensorget_tensor () const
 
const Shape & get_shape () const
 
const PartialShape & get_partial_shape () const
 
const element::Typeget_element_type () const
 
 Output (const Output &)=default
 
 Output (Output &&)=default
 
Outputoperator= (const Output &)=default
 

Protected Attributes

Node * m_node
 
size_t m_index
 
std::shared_ptr< Tensorm_tensor
 
std::vector< Input * > m_inputs
 

Constructor & Destructor Documentation

◆ Output()

ngraph::descriptor::Output::Output ( Node *  node,
size_t  index,
const std::shared_ptr< Tensor > &  tensor 
)
Parameters
nodeNode that owns this output.
indexPosition of the output tensor in all output tensors
tensorThe tensor where the value will be written

Member Function Documentation

◆ get_element_type()

const element::Type& ngraph::descriptor::Output::get_element_type ( ) const
Returns
the element type of the output

◆ get_partial_shape()

const PartialShape& ngraph::descriptor::Output::get_partial_shape ( ) const
Returns
the partial shape of the output

◆ get_shape()

const Shape& ngraph::descriptor::Output::get_shape ( ) const
Returns
the shape of the output

The documentation for this class was generated from the following file: