Searched refs:output_dims_for_index_array (Results 1 – 1 of 1) sorted by relevance
224 std::vector<int64> output_dims_for_index_array; in FoldGatherOfGather() local228 output_dims_for_index_array.push_back(gathered_index_components_seen); in FoldGatherOfGather()245 x, y, source_dim_for_index_array, output_dims_for_index_array, in FoldGatherOfGather()