Refactors serialize and deserialize value to switch/function pattern.
This is a pattern we have used frequently (e.g. in ReferenceResolvingExecutor's evaluate method) to persist more information in traces. Currently a little difficult to narrow down hotspots in serialization code. Additionally adds PyType annotations and cleans up the contracts a little bit, easier with this new factorization. PiperOrigin-RevId: 322246977
Showing
- tensorflow_federated/python/core/impl/executor_service_utils.py 211 additions, 161 deletions...flow_federated/python/core/impl/executor_service_utils.py
- tensorflow_federated/python/core/impl/executor_service_utils_test.py 11 additions, 18 deletions...federated/python/core/impl/executor_service_utils_test.py
Loading
Please register or sign in to comment