Home
last modified time | relevance | path

Searched refs:load_variable (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dbasic_session_run_hooks_test.py406 checkpoint_utils.load_variable(self.model_dir,
436 checkpoint_utils.load_variable(self.model_dir,
593 checkpoint_utils.load_variable(self.model_dir,
607 checkpoint_utils.load_variable(self.model_dir,
614 checkpoint_utils.load_variable(self.model_dir,
674 checkpoint_utils.load_variable(self.model_dir,
690 checkpoint_utils.load_variable(self.model_dir,
695 checkpoint_utils.load_variable(self.model_dir,
700 checkpoint_utils.load_variable(self.model_dir,
705 checkpoint_utils.load_variable(self.model_dir,
[all …]
H A Dcheckpoint_utils_test.py87 checkpoint_utils.load_variable(checkpoint_dir, "var1"), [])
95 checkpoint_utils.load_variable(checkpoint_dir, "var5"), [])
102 checkpoint_utils.load_variable(checkpoint_dir, "var1"), v1)
104 checkpoint_utils.load_variable(checkpoint_dir, "var2"), v2)
106 checkpoint_utils.load_variable(checkpoint_dir, "var3"), v3)
108 checkpoint_utils.load_variable(checkpoint_dir, "useful_scope/var4"), v4)
128 checkpoint_utils.load_variable(checkpoint_dir, "var1"), v1)
H A Dcheckpoint_utils.py72 def load_variable(ckpt_dir_or_file, name): function
H A Dtraining.py74 from tensorflow.python.training.checkpoint_utils import load_variable
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.train.pbtxt108 name: "load_variable"
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v2_checkpoint_test.py55 return checkpoint_utils.load_variable(self._get_tmpdir(name), variables[0])
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.train.pbtxt376 name: "load_variable"