Tensorflow 2nd order derivatives is None

Question

I using google colab and use tensorflow 1.15.2, my code below:

class PDENet:
  # Init
  def __init__(self, xb, yb, ub, x, y, layers):
    
    # Xb = np.concatena