Searched refs:output_detection_classes (Results 1 – 1 of 1) sorted by relevance
5344 ITensorProxyPtr output_detection_classes = layer->getOutput(3); in ConvertCombinedNMS() local5349 *output_detection_classes->trt_tensor()); in ConvertCombinedNMS()5351 output_detection_classes = layer_detection_classes->getOutput(0); in ConvertCombinedNMS()5364 params->outputs->push_back(TRT_TensorOrWeights(output_detection_classes)); in ConvertCombinedNMS()