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.