Searched defs:save_dir (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/ |
| H A D | checkpoint_management.py | 46 def _GetCheckpointFilename(save_dir, latest_filename): argument
63 def generate_checkpoint_state_proto(save_dir, argument
132 def update_checkpoint_state(save_dir, argument
174 def update_checkpoint_state_internal(save_dir, argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/repro/ |
| H A D | after_aot.py | 209 def generate_compiler_repro_string(gm, args, *, stable_output=False, save_dir=None): argument
264 save_dir=None, argument
348 save_dir=None, argument
749 save_dir=None, argument
|
| H A D | after_dynamo.py | 158 save_dir=None, argument
468 save_dir=None, argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | debug_utils.py | 514 def __init__(self, save_dir=None, *, pbar=None): argument
598 def __init__(self, save_dir, *, stable_hash=False): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | saver_test.py | 1315 all_model_checkpoint_paths, save_dir): argument
1681 all_model_checkpoint_paths, save_dir): argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | fx_minifier.py | 180 save_dir=None, argument
|
| /aosp_15_r20/external/autotest/server/ |
| H A D | site_linux_system.py | 396 def stop_capture(self, save_dir=None, save_filename=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | save_test.py | 80 save_dir, argument
|