Searched refs:output_indices_tensor (Results 1 – 1 of 1) sorted by relevance
3982 nvinfer1::ITensor* output_indices_tensor = layer->getOutput(1); in ConvertTopK() local3984 params->outputs->push_back(TRT_TensorOrWeights(output_indices_tensor)); in ConvertTopK()