Class TensorArray
Defined in File tensor_array.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class TensorArray : public mindspore::ops::BaseOperator
Assert defined TensorArray operator prototype of lite.
Public Functions
-
inline TensorArray()
Constructor.
-
void Init(bool dynamic_size, bool identical_element_shapes, const std::vector<int> &element_shape, int data_type)
Method to init the op’s attributes.
-
void set_dynamic_size(bool dynamic_size)
Method to set dynamic_size attributes.
- Parameters
dynamic_size – [in] Define the dynamic_size.
-
void set_identical_element_shapes(bool identical_element_shapes)
Method to set identical_element_shapes attributes.
- Parameters
identical_element_shapes – [in] Define the identical element shapes
-
void set_element_shape(const std::vector<int> &element_shape)
Method to set element_shape attributes.
- Parameters
element_shape – [in] Define the element shape.
-
void set_data_type(int data_type)
Method to set data_type attributes.
- Parameters
data_type – [in] Define the data type.
-
bool get_dynamic_size() const
Method to get dynamic_size attributes.
-
bool get_identical_element_shapes() const
Method to get element_shapes attributes.
-
const std::vector<int> get_element_shape() const
Method to get element_shape attributes.
-
int get_data_type() const
Method to get data_type attributes.
-
inline TensorArray()