Compile-time descriptor of a first-class value that is a tensor. More...
#include <tensor.hpp>
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::Type & | get_element_type () const |
| const Shape & | get_shape () const |
| const PartialShape & | get_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 |
| Node * | m_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 |
Compile-time descriptor of a first-class value that is a tensor.
