Class TensorListReserve

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class TensorListReserve : public mindspore::ops::BaseOperator

TensorListReserve defined TensorListReserve operator prototype of lite.

Public Functions

inline TensorListReserve()

Constructor.

void Init(const int64_t element_dtype, const int64_t shape_type)

Method to init the op's attributes.

参数

  • element_dtype – [in] Define the element data type in op;

  • shape_type – [in] Define the shape type in op;

void set_element_dtype(const int64_t element_dtype)

Method to set the op's element_dtype attributes.

参数

element_dtype – [in] Define the element data type in op;

void set_shape_type(const int64_t shape_type)

Method to set the op's shape_type attributes.

参数

shape_type – [in] Define the shape type in op;

int64_t get_element_dtype() const

Method to get the op's element_dtype attributes.

int64_t get_shape_type() const

Method to get the op's shape_type attributes.