Searched refs:TfTypeToTrtType (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.h | 171 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type);
|
| D | utils.cc | 206 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType() function
|
| D | convert_nodes_test.cc | 1054 TF_ASSERT_OK(TfTypeToTrtType(DataTypeToEnum<T>::v(), &trt_type)); in TestGetWeightRange()
1469 TF_RETURN_IF_ERROR(TfTypeToTrtType(data.tensor.dtype(), &trt_type)); in BuildAndRun()
1584 TF_ASSERT_OK(TfTypeToTrtType(DataTypeToEnum<T>::v(), &dtype)); in AddTestWeights()
1844 TF_ASSERT_OK(TfTypeToTrtType(tf_type, &trt_type));
4450 TF_ASSERT_OK(TfTypeToTrtType(tf_type_, &trt_type)); in TEST_P()
5916 TF_ASSERT_OK(TfTypeToTrtType(dtype, &trt_type)); in TEST_P()
6094 TF_ASSERT_OK(TfTypeToTrtType(dtype, &trt_type)); in TEST_P()
6568 TF_ASSERT_OK(TfTypeToTrtType(dtype, &trt_type)); in TEST_P()
6808 TF_ASSERT_OK(TfTypeToTrtType(tf_type_, &trt_type_)); in TEST_P()
|
| D | convert_nodes.cc | 222 TF_CHECK_OK(TfTypeToTrtType(this->at(key)->type(), &trt_dtype)); in get()
311 TF_RETURN_IF_ERROR(TfTypeToTrtType(dtype, trt_dtype)); in ValidateTensorProperties()
514 TF_RETURN_IF_ERROR(TfTypeToTrtType(dtype, &trt_type)); in CreateBroadcastableScalarConstant()
4320 TF_RETURN_IF_ERROR(TfTypeToTrtType(converted_dtype, &trt_dtype)); in TfTensorToTrtWeights()
6410 TF_RETURN_IF_ERROR(TfTypeToTrtType(tf_dtype, &trt_dtype)); in ConvertGraphDefToEngine()
|