Home
last modified time | relevance | path

Searched refs:save_function (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/saving/
H A Dsaveable_object_util.py476 def __init__(self, names_and_slices, save_function, restore_function, name): argument
477 self.save_function = save_function
485 tensors = save_function(name_tensor)
515 save_function=save_fn,
553 return factory(save_function=save_fn, restore_function=restore_fn,
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload.py303 save_fn_id = saveable_object_proto.save_function
311 save_fn_id = saveable_object_proto.save_function
H A Dsave.py363 saveable_object_proto.save_function = self.node_ids[save_fn]
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dsaved_object_graph.proto249 int32 save_function = 2; field
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dsaved_object_graph.pb.h2670 ::int32_t save_function() const; in Swap()
5203 inline ::int32_t SaveableObject::save_function() const { in save_function() function