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

Compile-time descriptor of a first-class value that is a tensor. More...

#include <tensor.hpp>

Collaboration diagram for ngraph::descriptor::Tensor:
Collaboration graph

Public Member Functions

 Tensor (const element::Type &element_type, const PartialShape &pshape, const std::string &name)
 Tensor (const element::Type &element_type, const PartialShape &pshape, Node *node, size_t node_output_number)
void set_names (const std::unordered_set< std::string > &names)
void set_tensor_type (const element::Type &element_type, const PartialShape &pshape)
void set_element_type (const element::Type &elemenet_type)
void set_partial_shape (const PartialShape &partial_shape)
void set_lower_value (const HostTensorPtr &value)
 sets lower bound value description
void set_upper_value (const HostTensorPtr &value)
 sets upper bound value description
void invalidate_values ()
 unsets bound value descriptions
const element::Typeget_element_type () const
const Shapeget_shape () const
const PartialShapeget_partial_shape () const
HostTensorPtr get_lower_value () const
 gets lower bound value description
HostTensorPtr get_upper_value () const
 gets upper bound value description
bool has_and_set_bound () const
 checks if lower and upper bound are set and point to the same HostTensor
size_t size () const

Protected Attributes

element::Type m_element_type
Shape m_shape
PartialShape m_partial_shape
Nodem_node {nullptr}
HostTensorPtr m_lower_value
HostTensorPtr m_upper_value
size_t m_node_output_number {0}
std::string m_name
std::unordered_set< std::string > m_names

Detailed Description

Compile-time descriptor of a first-class value that is a tensor.

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