Searched refs:_maybe_get_remote_value (Results 1 – 1 of 1) sorted by relevance
267 def _maybe_get_remote_value(val): function411 *nest.map_structure(_maybe_get_remote_value, replica_args),412 **nest.map_structure(_maybe_get_remote_value, replica_kwargs))