Searched refs:new_split_ops (Results 1 – 1 of 1) sorted by relevance
174 llvm::SmallVector<mlir::TF::SplitOp, 4> new_split_ops; in HandleTileShardedInputs() local175 new_split_ops.reserve(split_ops_for_tiled_input.size() * num_splits); in HandleTileShardedInputs()185 new_split_ops.emplace_back(child_split_op); in HandleTileShardedInputs()189 std::swap(new_split_ops, split_ops_for_tiled_input); in HandleTileShardedInputs()