Parallelising dataset on Tensorflow:throws error

Question

I have written the following function to define a CNN-model :

import pandas as pd
import numpy as np
from sklearn.preprocessing import LabelEncoder
from tenso