Home
last modified time | relevance | path

Searched refs:num_splits_and_index (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc158 for (auto num_splits_and_index : llvm::enumerate(dimension_splits)) { in HandleTileShardedInputs() local
159 const int num_splits = num_splits_and_index.value(); in HandleTileShardedInputs()
160 const int dimension_index = num_splits_and_index.index(); in HandleTileShardedInputs()