Searched refs:k_v (Results 1 – 2 of 2) sorted by relevance
49 auto k_v = GetValue<int>(input_args[1]->BuildValue()); in TopKInfer() local51 x_shape[ndims] = k_v; in TopKInfer()
2373 k_v = k['value']2374 validator.check_value_type('k', k_v, (int,), self.name)2377 x_shape[ndim] = k_v