Searched defs:trt_dtype (Results 1 – 3 of 3) sorted by relevance
97 inline Status ConvertDType(DataType tf_dtype, nvinfer1::DataType* trt_dtype) { in ConvertDType()176 nvinfer1::DataType trt_dtype(nvinfer1::DataType::kFLOAT); in get() local264 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() local645 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() local1583 nvinfer1::DataType trt_dtype; in BinaryTensorOpWeight() local[all …]
432 nvinfer1::DataType trt_dtype; member
99 DataType TrtDataTypeToTf(nvinfer1::DataType trt_dtype) { in TrtDataTypeToTf()1106 const nvinfer1::DataType trt_dtype = in CheckDataTypeMatches() local1196 nvinfer1::DataType trt_dtype = nvinfer1::DataType::kFLOAT) { in AddTestTensor()