SetOriginFormat

Function Usage

Sets the original data layout format of a tensor into CompileTimeTensorDesc.

Prototype

void SetOriginFormat(const ge::Format format)

Parameters

Parameter	Input/Output	Description
format	Input	Original data layout format to be set. For details about the definition of ge::Format, see Format.

Parameter

Input/Output

Description

format

Input

Original data layout format to be set.

For details about the definition of ge::Format, see Format.

Returns

None

Constraints

None

Examples

auto dtype_ = ge::DataType::DT_INT32;
StorageFormat fmt_(ge::Format::FORMAT_NC, ge::FORMAT_NCHW, {});
ExpandDimsType type_("1001");
gert::CompileTimeTensorDesc td;
td.SetDataType(dtype_);
auto dtype = td.GetDataType(); // ge::DataType::DT_INT32;
td.SetStorageFormat(fmt_.GetStorageFormat());
auto storage_fmt = td.GetStorageFormat(); // ge::FORMAT_NCHW
td.SetOriginFormat(fmt_.GetOriginFormat());
auto origin_fmt = td.GetOriginFormat(); // ge::Format::FORMAT_NC
td.SetExpandDimsType(type_);
auto type = td.GetExpandDimsType(); // type_("1001")

Parent topic: CompileTimeTensorDesc