bhack
bhack
The CI is failing
Probably you need to cherry-pick https://github.com/tensorflow/addons/pull/2580
@seanpmorgan you need to consent the bot for the CLA on this cherry-pick
Please check https://github.com/tensorflow/federated/issues/1254#issuecomment-962461866
@howl-anderson I don't think that he is still active in the project.
/cc @lgeiger Can you replicate this on your M1?
Have you already tried with https://www.tensorflow.org/api_docs/python/tf/config/set_soft_device_placement ?
@szutenberg Could it be a side effect of your introduced `var.device`?
@MR-T77 Yes sorry I missed, is that your code is not well formatted.