Searched refs:call_with_mapped_captures (Results 1 – 3 of 3) sorted by relevance
233 object_map, call_with_mapped_captures): argument280 call_with_mapped_captures)360 call_with_mapped_captures=None): argument383 call_with_mapped_captures=call_with_mapped_captures))412 call_with_mapped_captures=None): argument424 call_with_mapped_captures)
475 def create_saveable_object(factory, name, call_with_mapped_captures): argument491 if (call_with_mapped_captures is None or497 return call_with_mapped_captures(concrete_save_fn, [name])501 return call_with_mapped_captures(concrete_restore_fn, restored_tensors)
697 call_with_mapped_captures=functools.partial(