Class TensorArray

• Defined in File tensor_array.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Class Documentation

class TensorArray : public mindspore::ops::PrimitiveC

Assert defined TensorArray operator prototype of lite.

Public Functions

inline TensorArray()

Constructor.

~TensorArray() = default

Destructor.

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.