Searched refs:TrtTypeToTfType (Results 1 – 4 of 4) sorted by relevance
172 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type);
224 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType() function
1456 TF_ASSERT_OK(TrtTypeToTfType(trt_dtype, &tf_type)); in CheckDataTypeMatches()1536 TF_ASSERT_OK(TrtTypeToTfType(trt_type, &tf_type)); in AddTestTensorWithTFDims()
882 return TrtTypeToTfType(trt_type, tf_type); in GetTfType()1093 TF_CHECK_OK(TrtTypeToTfType(trt_dtype, &tf_dtype)); in GetTempWeights()