Searched refs:output_values_dims (Results 1 – 1 of 1) sorted by relevance
1467 auto output_values_dims = output_values.mutable_shape()->mutable_dims(); in ProcessTopkV2Operator() local1470 output_values_dims->push_back(input_values_shape.dims(dim)); in ProcessTopkV2Operator()1474 output_values_dims->push_back(k_value); in ProcessTopkV2Operator()