[Bug]: __step() method of WeightsBoundaryCalculator executes eagerly

[gfouilhe]

Module

None

Contact Details

No response

Current Behavior

The execution is eager. This is because mangled name isn't supported.

Expected Behavior

Execution in graph.

Version

v0.1.0

Environment

- OS:
- Python version:
- Tensorflow version:
- Packages used version:

Relevant log output

WARNING:tensorflow:AutoGraph could not transform <bound method WeightsBoundaryCalculator.__step of <tensorflow.python.eager.function.TfMethodTarget object at 0x7fba8c7b8a00>> and will run it as-is.
Cause: mangled names are not yet supported.

To Reproduce

just call WeightsBoundaryCalculator.__step