Searched refs:split_ops_for_tiled_input (Results 1 – 1 of 1) sorted by relevance
149 llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; in HandleTileShardedInputs() local150 split_ops_for_tiled_input.reserve( in HandleTileShardedInputs()164 if (split_ops_for_tiled_input.empty()) { in HandleTileShardedInputs()170 split_ops_for_tiled_input.emplace_back(root_split_op); in HandleTileShardedInputs()175 new_split_ops.reserve(split_ops_for_tiled_input.size() * num_splits); in HandleTileShardedInputs()177 for (auto split_op : split_ops_for_tiled_input) { in HandleTileShardedInputs()189 std::swap(new_split_ops, split_ops_for_tiled_input); in HandleTileShardedInputs()196 for (auto split_op : split_ops_for_tiled_input) in HandleTileShardedInputs()