Home
last modified time | relevance | path

Searched defs:trt_dtype (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType()
176 nvinfer1::DataType trt_dtype(nvinfer1::DataType::kFLOAT); in get() local
264 nvinfer1::DataType* trt_dtype, in ValidateTensorProperties()
322 string DebugString(const nvinfer1::DataType trt_dtype) { in DebugString()
450 const nvinfer1::DataType trt_dtype = trt_weights.type; in CreateConstantLayer() local
645 SimpleITensor(nvinfer1::DataType trt_dtype, const nvinfer1::Dims& trt_dims) in SimpleITensor()
658 void setType(nvinfer1::DataType trt_dtype) override { in setType()
707 TRT_TensorOrWeights::TRT_TensorOrWeights(nvinfer1::DataType trt_dtype, in TRT_TensorOrWeights()
921 nvinfer1::DataType trt_dtype; in ConvertToTensorOrWeights() local
1583 nvinfer1::DataType trt_dtype; in BinaryTensorOpWeight() local
[all …]
Dconvert_nodes.h432 nvinfer1::DataType trt_dtype; member
Dconvert_nodes_test.cc99 DataType TrtDataTypeToTf(nvinfer1::DataType trt_dtype) { in TrtDataTypeToTf()
1106 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local
1196 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor()