Searched defs:trt_type (Results 1 – 3 of 3) sorted by relevance
206 Status TfTypeToTrtType(DataType tf_type, nvinfer1::DataType* trt_type) { in TfTypeToTrtType()224 Status TrtTypeToTfType(nvinfer1::DataType trt_type, DataType* tf_type) { in TrtTypeToTfType()
1053 nvinfer1::DataType trt_type; in TestGetWeightRange() local1468 nvinfer1::DataType trt_type; in BuildAndRun() local1533 nvinfer1::DataType trt_type = nvinfer1::DataType::kFLOAT, in AddTestTensorWithTFDims()1843 nvinfer1::DataType trt_type; local4449 nvinfer1::DataType trt_type; in TEST_P() local5915 nvinfer1::DataType trt_type; in TEST_P() local6093 nvinfer1::DataType trt_type; in TEST_P() local6567 nvinfer1::DataType trt_type; in TEST_P() local
493 nvinfer1::DataType trt_type = nvinfer1::DataType::kINT32, in CreateScalarConstant()510 nvinfer1::DataType trt_type = nvinfer1::DataType::kFLOAT; // Default to FP32. in CreateBroadcastableScalarConstant() local881 nvinfer1::DataType trt_type = tensor()->getType(); in GetTfType() local