Problem with tf.while_loop InvalidArgumentError: Index out of range using input dim 0; input has only 0 dims [Op:StridedSlice] name: strided_slice/

Question

I\'m trying to parallelize this chunk of code using Tensorflow -

import numpy as np
import tensorflow as tf
import time
#Importing a generic dataset from Kera