Searched refs:source_dim_for_index_array (Results 1 – 1 of 1) sorted by relevance
225 int64_t source_dim_for_index_array = in FoldGatherOfGather() local227 CHECK_NE(source_dim_for_index_array, source->output_dims().size()); in FoldGatherOfGather()250 x, y, source_dim_for_index_array, output_dims_for_index_array, in FoldGatherOfGather()