SetDataType
Description
Sets the data type of a tensor into CompileTimeTensorDesc.
Prototype
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
Restrictions
None
Example
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