Searched refs:final_topk_shape (Results 1 – 1 of 1) sorted by relevance
1726 const Shape final_topk_shape = ShapeUtil::MakeTupleShape( in HandleSort() local1730 final_topk_shape, sort_dim, in HandleSort()