Похоже, что слой UpSampling2D в Keras не совместим с TPU в Google Collab. Есть ли возможность создать пользовательский слой, совместимый с TPU и выполняющий ту же операцию?