Searched refs:output_dims_for_index_array (Results 1 – 1 of 1) sorted by relevance
229 std::vector<int64_t> output_dims_for_index_array; in FoldGatherOfGather() local233 output_dims_for_index_array.push_back(gathered_index_components_seen); in FoldGatherOfGather()250 x, y, source_dim_for_index_array, output_dims_for_index_array, in FoldGatherOfGather()