Home
last modified time | relevance | path

Searched defs:checkpoint_path (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather_test.cc78 const auto checkpoint_path = in TestSinglePartition() local
291 const auto checkpoint_path = in TestMultiPartition() local
575 const auto checkpoint_path = in TestReadTensorSlice() local
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dsaver.py945 def _build_eager(self, checkpoint_path, build_save, build_restore): argument
949 def _build(self, checkpoint_path, build_save, build_restore): argument
1738 def object_graph_key_mapping(checkpoint_path): argument
1758 def saver_from_object_based_checkpoint(checkpoint_path, argument
H A Devaluation.py168 def _evaluate_once(checkpoint_path, argument
/aosp_15_r20/external/libopus/dnn/torch/osce/
H A Dexport_model_weights.py133 checkpoint_path = args.checkpoint variable
/aosp_15_r20/external/libopus/dnn/torch/lossgen/
H A Dtrain_lossgen.py95 checkpoint_path = os.path.join(checkpoint_dir, f'lossgen_{epoch}.pth') variable
/aosp_15_r20/external/libopus/dnn/torch/plc/
H A Dtrain_plc.py141 checkpoint_path = os.path.join(checkpoint_dir, f'plc{args.suffix}_{epoch}.pth') variable
/aosp_15_r20/external/tensorflow/tensorflow/python/tools/
H A Dsaved_model_aot_compile.py311 def aot_compile_cpu_meta_graph_def(checkpoint_path, argument
/aosp_15_r20/external/libopus/dnn/torch/fargan/
H A Dtrain_fargan.py164 checkpoint_path = os.path.join(checkpoint_dir, f'fargan{args.suffix}_{epoch}.pth') variable
H A Dadv_train_fargan.py266 checkpoint_path = os.path.join(checkpoint_dir, f'fargan{args.suffix}_adv_{epoch}.pth') variable
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Devent.proto79 string checkpoint_path = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/saving/
H A Dsaved_model_experimental.py234 mode, has_saved_vars, builder, model, custom_objects, checkpoint_path, argument
/aosp_15_r20/external/libopus/dnn/torch/rdovae/
H A Dtrain_rdovae.py274 checkpoint_path = os.path.join(checkpoint_dir, f'checkpoint_epoch_{epoch}.pth') variable
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dexample_query_plan_engine_test.cc153 std::string checkpoint_path) { in ReadTensors()
/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/util/
Devent.pb.h2316 inline const std::string& SessionLog::checkpoint_path() const { in checkpoint_path() function
2347 inline void SessionLog::set_allocated_checkpoint_path(std::string* checkpoint_path) { in set_allocated_checkpoint_path()
/aosp_15_r20/external/tensorflow/tensorflow/python/grappler/
H A Dlayout_optimizer_test.py239 def _train(self, checkpoint_path, layout_optimizer=False, restore=False): argument