MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / _trace_resource_initializers

Function _trace_resource_initializers

tensorflow/python/saved_model/save.py:472–489  ·  view source on GitHub ↗

Create concrete functions from `CapturableResource` objects.

(accessible_objects)

Source from the content-addressed store, hash-verified

470
471
472def _trace_resource_initializers(accessible_objects):
473 """Create concrete functions from `CapturableResource` objects."""
474 resource_initializers = []
475
476 def _wrap_initializer(obj):
477 obj._initialize() # pylint: disable=protected-access
478 return constant_op.constant(1.) # Dummy control output
479
480 def _wrap_obj_initializer(obj):
481 return lambda: _wrap_initializer(obj)
482
483 for obj in accessible_objects:
484 if isinstance(obj, tracking.CapturableResource):
485 resource_initializers.append(def_function.function(
486 _wrap_obj_initializer(obj),
487 # All inputs are captures.
488 input_signature=[]).get_concrete_function())
489 return resource_initializers
490
491
492_AssetInfo = collections.namedtuple(

Callers 1

_fill_meta_graph_defFunction · 0.85

Calls 4

_wrap_obj_initializerFunction · 0.85
appendMethod · 0.45
get_concrete_functionMethod · 0.45
functionMethod · 0.45

Tested by

no test coverage detected