Fixes two bugs in transformations infrastructure:
* Fixes `building_blocks.Block` traversal in `transform_preorder`. * Extends `unwrap_placement` to handle no-arg lambdas. PiperOrigin-RevId: 320601917
Showing
- tensorflow_federated/python/core/impl/compiler/transformation_utils.py 1 addition, 1 deletion...derated/python/core/impl/compiler/transformation_utils.py
- tensorflow_federated/python/core/impl/compiler/transformation_utils_test.py 15 additions, 0 deletions...ed/python/core/impl/compiler/transformation_utils_test.py
- tensorflow_federated/python/core/impl/compiler/tree_transformations.py 5 additions, 2 deletions...derated/python/core/impl/compiler/tree_transformations.py
- tensorflow_federated/python/core/impl/compiler/tree_transformations_test.py 11 additions, 0 deletions...ed/python/core/impl/compiler/tree_transformations_test.py
Loading
Please register or sign in to comment