Searched refs:gather_dim (Results 1 – 1 of 1) sorted by relevance
208 for (int64 gather_dim : source->output_dims()) { in FoldGatherOfGather() local209 simulated_index.insert(simulated_index.begin() + gather_dim, in FoldGatherOfGather()