| /external/tensorflow/tensorflow/python/training/ |
| saver_large_variable_test.py | 23 from tensorflow.core.protobuf import saver_pb2
52 }, write_version=saver_pb2.SaverDef.V1)
|
| checkpoint_management.py | 29 from tensorflow.core.protobuf import saver_pb2
317 if format_version == saver_pb2.SaverDef.V2:
340 saver_pb2.SaverDef.V2)
342 saver_pb2.SaverDef.V1)
371 saver_pb2.SaverDef.V2)
413 saver_pb2.SaverDef.V2)
427 checkpoint_format_version=saver_pb2.SaverDef.V2,
441 if checkpoint_format_version == saver_pb2.SaverDef.V2:
|
| saver.py | 33 from tensorflow.core.protobuf import saver_pb2
84 def __init__(self, write_version=saver_pb2.SaverDef.V2):
113 if self._write_version == saver_pb2.SaverDef.V1:
119 elif self._write_version == saver_pb2.SaverDef.V2:
293 if self._write_version == saver_pb2.SaverDef.V2:
528 return saver_pb2.SaverDef(
553 return saver_pb2.SaverDef(
697 write_version=saver_pb2.SaverDef.V2,
[all...] |
| checkpoint_management_test.py | 28 from tensorflow.core.protobuf import saver_pb2
278 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
297 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
312 for version in (saver_pb2.SaverDef.V2, saver_pb2.SaverDef.V1):
|
| saver_test.py | 36 from tensorflow.core.protobuf import saver_pb2
386 for ver in (saver_pb2.SaverDef.V1, saver_pb2.SaverDef.V2):
[all...] |
| training.py | 118 from tensorflow.core.protobuf.saver_pb2 import *
|
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| checkpoint_convert.py | 47 from tensorflow.core.protobuf import saver_pb2
242 write_version = (saver_pb2.SaverDef.V1
243 if write_v1_checkpoint else saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| functional_saver.py | 21 from tensorflow.core.protobuf import saver_pb2
70 return saver_pb2.SaverDef(
74 version=saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/python/tools/ |
| freeze_graph.py | 47 from tensorflow.core.protobuf import saver_pb2
91 checkpoint_version=saver_pb2.SaverDef.V2):
117 checkpoint_version: Tensorflow variable file format (saver_pb2.SaverDef.V1
118 or saver_pb2.SaverDef.V2)
280 saver_def = saver_pb2.SaverDef()
303 checkpoint_version=saver_pb2.SaverDef.V2):
330 checkpoint_version: Tensorflow variable file format (saver_pb2.SaverDef.V1
331 or saver_pb2.SaverDef.V2).
368 checkpoint_version = saver_pb2.SaverDef.V1
370 checkpoint_version = saver_pb2.SaverDef.V
[all...] |
| freeze_graph_test.py | 26 from tensorflow.core.protobuf import saver_pb2
163 self._testFreezeGraph(saver_pb2.SaverDef.V1)
167 self._testFreezeGraph(saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| make_test_graphs.py | 25 from tensorflow.core.protobuf import saver_pb2
54 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
69 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| exporter_test.py | 29 from tensorflow.core.protobuf import saver_pb2
109 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| evaluation_test.py | 31 from tensorflow.core.protobuf import saver_pb2
250 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| variables.py | 28 from tensorflow.core.protobuf import saver_pb2
748 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| builder_impl.py | 29 from tensorflow.core.protobuf import saver_pb2
222 write_version=saver_pb2.SaverDef.V2,
[all...] |
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| meta_graph_transform_test.py | 28 from tensorflow.core.protobuf import saver_pb2
139 saver_def = saver_pb2.SaverDef()
|
| /external/tensorflow/tensorflow/python/framework/ |
| meta_graph.py | 34 from tensorflow.core.protobuf import saver_pb2
606 if saver_def and not isinstance(saver_def, saver_pb2.SaverDef):
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_ops_test.py | 30 from tensorflow.core.protobuf import saver_pb2
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| layout_optimizer_test.py | 26 from tensorflow.core.protobuf import saver_pb2
238 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| normalization_test.py | 24 from tensorflow.core.protobuf import saver_pb2
64 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
[all...] |
