Searched refs:output_size_per_class (Results 1 – 1 of 1) sorted by relevance
5925 TRT_ShapedWeights output_size_per_class = inputs.at(2).weights(); in ConvertCombinedNMS() local5955 if (output_size_per_class.shape_.nbDims != 1) { in ConvertCombinedNMS()5961 *(static_cast<int*>(output_size_per_class.GetValues())); in ConvertCombinedNMS()