Rohit Singh

weight normalization layers throws error with TPU Strategy

4

Environment : Google Colab Hardware Accelerator: TPU TensorFlow version: 2.2.0-rc3 TensorFlow Addons: 0.9.1 **Describe the bug** Weight Normalization layer (tfa.layers.WeightNormalization) throws error when running in TPU **Code to reproduce the...