SetDataType
Function Usage
Sets the data type of a tensor into CompileTimeTensorDesc.
Prototype
1 | void SetDataType(const ge::DataType data_type) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
data_type |
Input |
Data type of the tensor to be set described by CompileTimeTensorDesc. For details about ge::DataType see DataType. |
Returns
None
Constraints
None
Examples
1 2 3 4 5 6 7 8 9 10 11 12 | 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