Home
last modified time | relevance | path

Searched refs:output_num_detections (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6068 nvinfer1::ITensor* output_num_detections = layer->getOutput(0); in ConvertCombinedNMS() local
6075 params->outputs->push_back(TRT_TensorOrWeights(output_num_detections)); in ConvertCombinedNMS()