Searched refs:output_indexes_dims (Results 1 – 1 of 1) sorted by relevance
1466 auto output_indexes_dims = output_indexes.mutable_shape()->mutable_dims(); in ProcessTopkV2Operator() local1469 output_indexes_dims->push_back(input_values_shape.dims(dim)); in ProcessTopkV2Operator()1473 output_indexes_dims->push_back(k_value); in ProcessTopkV2Operator()