Default MaxPoolingOp only supports NHWC on device type CPU
[[node functional_4/max_pooling2d_5/MaxPool (defined at :3) ]] [Op:__inference_train_function_2778]
Function call stack:
train_function
I am not able to resolve this problem, please help @Mubsi