Overview
CompileTimeTensorDesc is used to describe the basic information about a tensor during compilation, including the data types (ge::DataType), data layout formats (gert::StorageFormat, including the original format and runtime format), and dimension expansion rules during conversion from the original format to the runtime format. This class provides methods for obtaining and setting the tensor data type, and supports management of the original format and runtime format.
Header Files to Be Included
1 | #include <compute_node_info.h> |
Public Member Functions
CompileTimeTensorDesc() ge::DataType GetDataType() const const StorageFormat& GetFormat() const ge::Format GetOriginFormat() const ge::Format GetStorageFormat() const ExpandDimsType GetExpandDimsType() const void SetDataType(const ge::DataType data_type) void SetStorageFormat(const ge::Format format) void SetOriginFormat(const ge::Format format) void SetExpandDimsType(const ExpandDimsType &expand_dims_type)
Parent topic: CompileTimeTensorDesc