RemoteCall: Cache function handles.
Currently, whenever a functional_ops.remote_call(...) is executed against a remote worker, the function will be instantiated each and every time against the remote worker causing a memory leak on both the caller and the callee. Instead, we cache the function handles and reuse them. PiperOrigin-RevId: 188093266
Loading
Please sign in to comment