Hello

print(met.metrics_report())

According to the metrics reported printed after an XLA training session I observe that the functions cannot be processed in TPU.

Counter: aten::_linalg_svd Value: 3276 Counter: aten::_local_scalar_dense Value: 16380 Counter: aten::_unique2 Value: 3276 Counter: aten::bincount Value: 3276 

What should I do?