Searched refs:output_indices_tensor (Results 1 – 1 of 1) sorted by relevance
5716 nvinfer1::ITensor* output_indices_tensor = layer->getOutput(1); in ConvertArgMinMax() local5723 output_indices_tensor, &input_dims, params, &output_tensor)); in ConvertArgMinMax()5772 nvinfer1::ITensor* output_indices_tensor = layer->getOutput(1); in ConvertTopK() local5774 params->outputs->push_back(TRT_TensorOrWeights(output_indices_tensor)); in ConvertTopK()