Searched refs:trt_dtype (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | weights.cc | 102 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights() argument
105 : tensor_proxy_ptr_(new SimpleITensor(trt_dtype, trt_dims)), in TRT_TensorOrWeights()
190 nvinfer1::DataType trt_dtype, const DimsAdapter& dims) { in GetTempWeights() argument
192 TF_RETURN_IF_ERROR(TrtTypeToTfType(trt_dtype, &tf_dtype)); in GetTempWeights()
198 TRT_ShapedWeights::CreateWithTensor(trt_dtype, dims, tensor); in GetTempWeights()
|
| D | utils.cc | 55 string DebugString(const nvinfer1::DataType trt_dtype) { in DebugString() argument
56 switch (trt_dtype) { in DebugString()
|
| D | weights.h | 198 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,
|
| D | convert_nodes.cc | 223 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties() argument
226 TF_RETURN_IF_ERROR(TfTypeToTrtType(dtype, trt_dtype)); in ValidateTensorProperties()
934 nvinfer1::DataType trt_dtype; in ConvertToTensorOrWeights() local
939 /*validation_only_=*/true, &trt_dtype, &trt_dims, &batch_size)); in ConvertToTensorOrWeights()
944 *tensor_or_weights = TRT_TensorOrWeights(trt_dtype, trt_dims, batch_size); in ConvertToTensorOrWeights()
1193 tensor->setType(output.trt_dtype); in RenameAndMarkOutputTensors()
1196 << " with data type " << DebugString(output.trt_dtype) in RenameAndMarkOutputTensors()
3446 nvinfer1::DataType trt_dtype; in TfTensorToTrtWeights() local
3447 TF_RETURN_IF_ERROR(TfTypeToTrtType(converted_dtype, &trt_dtype)); in TfTensorToTrtWeights()
3451 *weights = TRT_ShapedWeights(trt_dtype); in TfTensorToTrtWeights()
[all …]
|
| D | utils.h | 104 string DebugString(const nvinfer1::DataType trt_dtype);
|
| D | convert_nodes.h | 262 nvinfer1::DataType trt_dtype; member
|
| D | convert_nodes_test.cc | 1257 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local
1260 TF_ASSERT_OK(TrtTypeToTfType(trt_dtype, &tf_type)); in CheckDataTypeMatches()
1385 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor() argument
1389 AddTestTensorWithTFDims(name, dims_vec, trt_dtype); in AddTestTensor()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/utils/ |
| D | trt_tensor_proxy.h | 39 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() argument
40 : trt_dtype_(trt_dtype), trt_dims_(trt_dims) {} in SimpleITensor()
63 void setType(nvinfer1::DataType trt_dtype) { trt_dtype_ = trt_dtype; } in setType() argument
126 explicit ITensorProxy(nvinfer1::DataType trt_dtype, in ITensorProxy() argument
129 new SimpleITensor(trt_dtype, trt_dims))), in ITensorProxy()
|