Home
last modified time | relevance | path

Searched refs:TFRT_DTYPE (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/utils/
Dtensor_util.cc89 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTFDTypeToTFRTDType() argument
91 return DType(DType::TFRT_DTYPE); in ConvertTFDTypeToTFRTDType()
101 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTFRTDTypeToTFDType() argument
102 case DType::TFRT_DTYPE: \ in ConvertTFRTDTypeToTFDType()
Dutils.cc58 #define DTYPE(TFRT_DTYPE, TF_DTYPE) \ in ConvertTfDTypeToTfrtDType() argument
60 return DType(DType::TFRT_DTYPE); in ConvertTfDTypeToTfrtDType()