Searched refs:split_ops_for_tiled_input (Results 1 – 1 of 1) sorted by relevance
143 llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; in HandleTileShardedInputs() local144 split_ops_for_tiled_input.reserve( in HandleTileShardedInputs()158 if (split_ops_for_tiled_input.empty()) { in HandleTileShardedInputs()164 split_ops_for_tiled_input.emplace_back(root_split_op); in HandleTileShardedInputs()169 new_split_ops.reserve(split_ops_for_tiled_input.size() * num_splits); in HandleTileShardedInputs()171 for (auto split_op : split_ops_for_tiled_input) { in HandleTileShardedInputs()183 std::swap(new_split_ops, split_ops_for_tiled_input); in HandleTileShardedInputs()190 for (auto split_op : split_ops_for_tiled_input) in HandleTileShardedInputs()