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) | |
| const std::string & | get_name () const |
| void | set_name (const std::string &name) |
| 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) |
| const element::Type & | get_element_type () const |
| const Shape & | get_shape () const |
| const PartialShape & | get_partial_shape () const |
| void | set_pool_offset (size_t) |
| size_t | get_pool_offset () const |
| size_t | size () const |
Protected Attributes | |
| element::Type | m_element_type |
| Shape | m_shape |
| PartialShape | m_partial_shape |
| Node * | m_node {nullptr} |
| size_t | m_node_output_number {0} |
| std::string | m_name |
| size_t | m_pool_offset {0} |
Compile-time descriptor of a first-class value that is a tensor.
