Searched refs:layer_detection_classes (Results 1 – 1 of 1) sorted by relevance
5347 nvinfer1::IIdentityLayer* layer_detection_classes = in ConvertCombinedNMS() local5350 layer_detection_classes->setOutputType(0, nvinfer1::DataType::kFLOAT); in ConvertCombinedNMS()5351 output_detection_classes = layer_detection_classes->getOutput(0); in ConvertCombinedNMS()