Home
last modified time | relevance | path

Searched refs:TfTypeToTrtType (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dutils.h171 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type);
Dutils.cc206 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType() function
Dconvert_nodes_test.cc1054 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()
Dconvert_nodes.cc222 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()