Skip to content
Snippets Groups Projects
Commit e90385b1 authored by Keith Rush's avatar Keith Rush Committed by tensorflow-copybara
Browse files

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
parent db837e9c
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment