Searched refs:trt_dtype (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.cc | 102 string DebugString(const nvinfer1::DataType trt_dtype) { in DebugString() argument
103 switch (trt_dtype) { in DebugString()
|
| D | convert_nodes.cc | 221 nvinfer1::DataType trt_dtype(nvinfer1::DataType::kFLOAT); in get() local
222 TF_CHECK_OK(TfTypeToTrtType(this->at(key)->type(), &trt_dtype)); in get()
223 return trt_dtype; in get()
308 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties() argument
311 TF_RETURN_IF_ERROR(TfTypeToTrtType(dtype, trt_dtype)); in ValidateTensorProperties()
766 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor() argument
767 : trt_dtype_(trt_dtype), trt_dims_(trt_dims) {} in SimpleITensor()
779 void setType(nvinfer1::DataType trt_dtype) override { in setType() argument
780 trt_dtype_ = trt_dtype; in setType()
838 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights() argument
[all …]
|
| D | convert_nodes.h | 277 explicit TRT_TensorOrWeights(nvinfer1::DataType trt_dtype,
453 nvinfer1::DataType trt_dtype; member
|
| D | utils.h | 105 string DebugString(const nvinfer1::DataType trt_dtype);
|
| D | convert_nodes_test.cc | 1453 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local
1456 TF_ASSERT_OK(TrtTypeToTfType(trt_dtype, &tf_type)); in CheckDataTypeMatches()
1562 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor() argument
1566 AddTestTensorWithTFDims(name, dims_with_batch, trt_dtype); in AddTestTensor()
|