Searched refs:ckpt_file (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 311 ckpt_file = _get_checkpoint_filename(ckpt_dir_or_file)
349 _set_variable_or_list_initializer(var, ckpt_file, tensor_name_in_ckpt)
391 _set_variable_or_list_initializer(var, ckpt_file, full_tensor_name)
404 ckpt_file, argument
427 ckpt_file, [tensor_name], [slice_spec], [base_type], name=name)[0]
445 def _set_variable_or_list_initializer(variable_or_list, ckpt_file, argument
471 _set_checkpoint_initializer(v, ckpt_file, tensor_name, slice_info.spec)
473 _set_checkpoint_initializer(variable_or_list, ckpt_file, tensor_name, "")
|
| D | checkpoint_management_test.py | 240 ckpt_file = open(ckpt_path, "w")
241 ckpt_file.write("")
242 ckpt_file.close()
250 ckpt_file = open(ckpt_path, "w")
251 ckpt_file.write("""
256 ckpt_file.close()
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | make_test_graphs.py | 80 ckpt_file = os.path.join(out_dir, 'test_graph_tfadd_with_ckpt_saver.ckpt')
81 saver.save(sess, ckpt_file)
|