Searched refs:broadcast_dims_vector (Results 1 – 1 of 1) sorted by relevance
1321 std::vector<int64> broadcast_dims_vector; in ReshardWithCollectivePermute() local1324 broadcast_dims_vector.push_back(i); in ReshardWithCollectivePermute()1328 sharding(), broadcast_dims_vector) == in ReshardWithCollectivePermute()1330 target, broadcast_dims_vector)) { in ReshardWithCollectivePermute()