Home
last modified time | relevance | path

Searched refs:DLDataTypeCode (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/
DDLConvertor.cpp23 dtype.code = DLDataTypeCode::kDLUInt; in getDLDataType()
26 dtype.code = DLDataTypeCode::kDLInt; in getDLDataType()
30 dtype.code = DLDataTypeCode::kDLFloat; in getDLDataType()
33 dtype.code = DLDataTypeCode::kDLFloat; in getDLDataType()
37 dtype.code = DLDataTypeCode::kDLInt; in getDLDataType()
40 dtype.code = DLDataTypeCode::kDLInt; in getDLDataType()
43 dtype.code = DLDataTypeCode::kDLInt; in getDLDataType()
46 dtype.code = DLDataTypeCode::kDLFloat; in getDLDataType()
49 dtype.code = DLDataTypeCode::kDLBool; in getDLDataType()
52 dtype.code = DLDataTypeCode::kDLComplex; in getDLDataType()
[all …]
Ddlpack.h133 } DLDataTypeCode; typedef
/external/tensorflow/tensorflow/c/eager/
Ddlpack.cc84 dtype.code = DLDataTypeCode::kDLFloat; in GetDlDataType()
90 dtype.code = DLDataTypeCode::kDLInt; in GetDlDataType()
97 dtype.code = DLDataTypeCode::kDLUInt; in GetDlDataType()
100 dtype.code = DLDataTypeCode::kDLBfloat; in GetDlDataType()
104 dtype.code = DLDataTypeCode::kDLComplex; in GetDlDataType()
158 case DLDataTypeCode::kDLUInt: in TfDataTypeFormDlDataType()
177 case DLDataTypeCode::kDLInt: in TfDataTypeFormDlDataType()
196 case DLDataTypeCode::kDLFloat: in TfDataTypeFormDlDataType()
212 case DLDataTypeCode::kDLBfloat: in TfDataTypeFormDlDataType()
222 case DLDataTypeCode::kDLComplex: in TfDataTypeFormDlDataType()