Class TensorListFromTensor
Defined in File tensor_list_from_tensor.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class TensorListFromTensor : public mindspore::ops::BaseOperator
TensorListFromTensor defined TensorListFromTensor operator prototype of lite.
Public Functions
-
inline TensorListFromTensor()
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.
-
inline TensorListFromTensor()