Tpu Layers Not Sticking