Searched refs:dimension_splits (Results 1 – 1 of 1) sorted by relevance
151 const auto& dimension_splits = input_sharding.tile_assignment_dimensions(); in HandleTileShardedInputs() local152 for (auto num_splits_and_index : llvm::enumerate(dimension_splits)) { in HandleTileShardedInputs()