Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc165 mlir::TF::SplitOp root_split_op; in HandleTileShardedInputs() local
167 original_source, builder, &root_split_op); in HandleTileShardedInputs()
170 split_ops_for_tiled_input.emplace_back(root_split_op); in HandleTileShardedInputs()