Searched refs:dim_sizes_for_composed_index (Results 1 – 1 of 1) sorted by relevance
235 std::vector<int64> dim_sizes_for_composed_index; in FoldGatherOfGather() local239 dim_sizes_for_composed_index.push_back(shape.dimensions(i)); in FoldGatherOfGather()247 dim_sizes_for_composed_index)); in FoldGatherOfGather()