Searched refs:k_w (Results 1 – 1 of 1) sorted by relevance
5754 TRT_ShapedWeights k_w = inputs.at(1).weights(); in ConvertTopK() local5755 if (k_w.count() != 1) { in ConvertTopK()5764 const int k = *(static_cast<int*>(k_w.GetValues())); in ConvertTopK()