Change the default partitioning method for gather and scatter to kExplicitBatch.
The default method is preferred and always has zero cost. We move BatchedGatherScatterNormalizer after partitioner, we will encounter more cases with explicit batch dims, compared with the implicit batch dims.
Change the default partitioning method for gather and scatter to kExplicitBatch.
The default method is preferred and always has zero cost. We move
BatchedGatherScatterNormalizerafter partitioner, we will encounter more cases with explicit batch dims, compared with the implicit batch dims.