Searched refs:split_ops_for_tiled_input (Results 1 – 1 of 1) sorted by relevance
147 llvm::SmallVector<mlir::TF::SplitOp, 4> split_ops_for_tiled_input; in HandleTileShardedInputs() local148 split_ops_for_tiled_input.reserve( in HandleTileShardedInputs()162 if (split_ops_for_tiled_input.empty()) { in HandleTileShardedInputs()168 split_ops_for_tiled_input.emplace_back(root_split_op); in HandleTileShardedInputs()173 new_split_ops.reserve(split_ops_for_tiled_input.size() * num_splits); in HandleTileShardedInputs()175 for (auto split_op : split_ops_for_tiled_input) { in HandleTileShardedInputs()187 std::swap(new_split_ops, split_ops_for_tiled_input); in HandleTileShardedInputs()194 for (auto split_op : split_ops_for_tiled_input) in HandleTileShardedInputs()