下载
中文
注册

GetFormat

函数功能

获取编译时Tensor的数据排布格式。

函数原型

const StorageFormat &GetFormat() const

参数说明

无。

返回值说明

返回数据排布格式。StorageFormat类型,包括原始Format和存储format。

约束说明

调用示例

StorageFormat storageFormat(ge::Format::FORMAT_NC, ge::FORMAT_NCHW, {});
gert::CompileTimeTensorDesc compileTimeTensorDesc;
compileTimeTensorDesc.SetStorageFormat(storageFormat.GetStorageFormat());
auto storage_fmt = compileTimeTensorDesc.GetFormat(); // ge::FORMAT_NCHW
compileTimeTensorDesc.SetOriginFormat(storageFormat.GetOriginFormat());
auto origin_fmt = compileTimeTensorDesc.GetOriginFormat(); // ge::Format::FORMAT_NC