Searched defs:trt_type (Results 1 – 4 of 4) sorted by relevance
153 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType()176 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType()
164 nvinfer1::DataType trt_type = tensor()->getType(); in GetTfType() local
852 nvinfer1::DataType trt_type; in TestGetWeightRange() local1272 nvinfer1::DataType trt_type; in BuildAndRun() local1345 nvinfer1::DataType trt_type = nvinfer1::DataType::kFLOAT, in AddTestTensorWithTFDims()1851 nvinfer1::DataType trt_type; local4225 nvinfer1::DataType trt_type; in TEST_P() local6143 nvinfer1::DataType trt_type; in TEST_P() local7975 nvinfer1::DataType trt_type; in TestConvertSplit() local
493 nvinfer1::DataType trt_type = nvinfer1::DataType::kINT32, in CreateScalarConstant()510 nvinfer1::DataType trt_type = nvinfer1::DataType::kFLOAT; // Default to FP32. in CreateBroadcastableScalarConstant() local