unable to train on complete batch in tensorflow in google collab

Question

I was trying to train a convolutional neural network to classify cats vs dogs in google collab here is the code I am using:

import tensorflow as tf
from tenso