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()
|