Searched refs:saver_module (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/training/ |
D | saver_test.py | 73 from tensorflow.python.training import saver as saver_module unknown 106 save = saver_module.Saver( 130 save = saver_module.Saver({"v0": v0, "v1": v1, "v2": v2.saveable}) 158 save2 = saver_module.Saver({"v0": v0_2, "v1": v1_2, "v2": v2_2.saveable}) 181 saver_module.Saver({"v0": v}).build() 199 saver_module.Saver() 210 save = saver_module.Saver([v1, v2]) 230 graph_saver = saver_module.Saver([w1, w2]) 241 graph_saver = saver_module.Saver([w1, w2]) 256 graph_saver = saver_module.Saver([w3, w4]) [all …]
|
D | saver_test_utils.py | 25 from tensorflow.python.training import saver as saver_module unknown 75 class CustomSaveable(saver_module.BaseSaverBuilder.SaveableObject): 81 saver_module.BaseSaverBuilder.SaveSpec(tensors[0], "", 83 saver_module.BaseSaverBuilder.SaveSpec(tensors[1], "",
|
D | checkpoint_management_test.py | 39 from tensorflow.python.training import saver as saver_module unknown 82 saver = saver_module.Saver(sharded=False) 92 saver = saver_module.Saver(sharded=True) 98 saver = saver_module.Saver(sharded=True) 123 save = saver_module.Saver({"v0": v0}) 138 save = saver_module.Saver({"v0": v0}) 282 saver = saver_module.Saver(sharded=sharded, write_version=version) 301 saver = saver_module.Saver(write_version=version) 316 saver = saver_module.Saver(sharded=sharded, write_version=version)
|
D | quantize_training_test.py | 32 from tensorflow.python.training import saver as saver_module unknown 69 saver = saver_module.Saver({'b': b})
|