Does any newer versions of 2.2 or higher support TPU with float64? I'm using VGP which requires float64 for numeric stability. i get the following error:
TypeError: Value passed to parameter 'input' has DataType float64 not in list of allowed values: bfloat16, float32, int32, uint32
Current TF does not based the documentation