Searched refs:ckpt_file (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 197 ckpt_file = _get_checkpoint_filename(ckpt_dir_or_file)
235 _set_variable_or_list_initializer(var, ckpt_file, tensor_name_in_ckpt)
277 _set_variable_or_list_initializer(var, ckpt_file, full_tensor_name)
290 ckpt_file, argument
313 ckpt_file, [tensor_name], [slice_spec], [base_type], name=name)[0]
331 def _set_variable_or_list_initializer(variable_or_list, ckpt_file, argument
357 _set_checkpoint_initializer(v, ckpt_file, tensor_name, slice_info.spec)
359 _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 | 74 ckpt_file = os.path.join(out_dir, 'test_graph_tfadd_with_ckpt_saver.ckpt')
75 saver.save(sess, ckpt_file)
|