Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc49 mlir::LogicalResult CreateSplitOp(const int num_split, in CreateSplitOp() function
166 auto result = CreateSplitOp(num_splits, dimension_index, location, in HandleTileShardedInputs()
181 CreateSplitOp(num_splits, dimension_index, location, in HandleTileShardedInputs()