| /external/tensorflow/tensorflow/python/training/ |
| H A D | saver_large_variable_test.py | 23 from tensorflow.core.protobuf import saver_pb2 namespace
52 }, write_version=saver_pb2.SaverDef.V1)
|
| H A D | saver.py | 34 from tensorflow.core.protobuf import saver_pb2 namespace
202 def __init__(self, write_version=saver_pb2.SaverDef.V2):
231 if self._write_version == saver_pb2.SaverDef.V1:
237 elif self._write_version == saver_pb2.SaverDef.V2:
410 if self._write_version == saver_pb2.SaverDef.V2:
810 return saver_pb2.SaverDef(
821 return saver_pb2.SaverDef(
1176 write_version=saver_pb2.SaverDef.V2,
1359 if not isinstance(self.saver_def, saver_pb2.SaverDef):
1360 raise ValueError("saver_def must be a saver_pb2
[all...] |
| H A D | saver_test.py | 38 from tensorflow.core.protobuf import saver_pb2 namespace
341 for ver in (saver_pb2.SaverDef.V1, saver_pb2.SaverDef.V2):
749 _WRITE_VERSION = saver_pb2.SaverDef.V1
782 if save._write_version is saver_pb2.SaverDef.V1:
789 if save._write_version is saver_pb2.SaverDef.V1:
867 if save._write_version is saver_pb2.SaverDef.V1:
878 if save._write_version is saver_pb2.SaverDef.V1:
1023 _WRITE_VERSION = saver_pb2.SaverDef.V2
1220 if save._write_version is saver_pb2
[all...] |
| H A D | training.py | 198 from tensorflow.core.protobuf.saver_pb2 import *
239 "SaverDef", # From saver_pb2.
|
| /external/tensorflow/tensorflow/contrib/rnn/python/tools/ |
| H A D | checkpoint_convert.py | 47 from tensorflow.core.protobuf import saver_pb2 namespace
242 write_version = (saver_pb2.SaverDef.V1
243 if write_v1_checkpoint else saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/python/tools/ |
| H A D | freeze_graph.py | 46 from tensorflow.core.protobuf import saver_pb2 namespace
76 checkpoint_version=saver_pb2.SaverDef.V2):
203 saver_def = saver_pb2.SaverDef()
226 checkpoint_version=saver_pb2.SaverDef.V2):
261 checkpoint_version = saver_pb2.SaverDef.V1
263 checkpoint_version = saver_pb2.SaverDef.V2
|
| H A D | freeze_graph_test.py | 25 from tensorflow.core.protobuf import saver_pb2 namespace
161 self._testFreezeGraph(saver_pb2.SaverDef.V1)
164 self._testFreezeGraph(saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| H A D | make_test_graphs.py | 25 from tensorflow.core.protobuf import saver_pb2 namespace
52 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
67 saver = saver_lib.Saver(name='abcprefix', write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| H A D | builder_impl.py | 28 from tensorflow.core.protobuf import saver_pb2 namespace
299 write_version=saver_pb2.SaverDef.V2,
387 write_version=saver_pb2.SaverDef.V2,
|
| /external/tensorflow/tensorflow/contrib/session_bundle/ |
| H A D | exporter_test.py | 29 from tensorflow.core.protobuf import saver_pb2 namespace
109 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| H A D | variables.py | 28 from tensorflow.core.protobuf import saver_pb2 namespace
689 write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/ |
| H A D | evaluation_test.py | 32 from tensorflow.core.protobuf import saver_pb2 namespace
239 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| H A D | cudnn_rnn_ops_test.py | 28 from tensorflow.core.protobuf import saver_pb2 namespace
217 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
269 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
313 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/contrib/meta_graph_transform/ |
| H A D | meta_graph_transform_test.py | 28 from tensorflow.core.protobuf import saver_pb2 namespace
139 saver_def = saver_pb2.SaverDef()
|
| /external/tensorflow/tensorflow/python/framework/ |
| H A D | meta_graph.py | 33 from tensorflow.core.protobuf import saver_pb2 namespace
556 if saver_def and not isinstance(saver_def, saver_pb2.SaverDef):
|
| /external/tensorflow/tensorflow/python/grappler/ |
| H A D | layout_optimizer_test.py | 26 from tensorflow.core.protobuf import saver_pb2 namespace
234 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
|
| /external/tensorflow/tensorflow/python/layers/ |
| H A D | normalization_test.py | 24 from tensorflow.core.protobuf import saver_pb2 namespace
62 saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V2)
|