| /external/tensorflow/tensorflow/python/saved_model/ |
| D | save_test.py | 55 from tensorflow.python.saved_model import save_options
514 save_options.VariablePolicy.SAVE_VARIABLE_DEVICES, True),
515 ("_DiscardDevices_ExportMetaGraph", save_options.VariablePolicy.NONE,
517 save_options.VariablePolicy.SAVE_VARIABLE_DEVICES, False),
518 ("_DiscardDevices_Save", save_options.VariablePolicy.NONE, False))
536 options = save_options.SaveOptions(
556 if save_devices == save_options.VariablePolicy.SAVE_VARIABLE_DEVICES:
573 if save_devices == save_options.VariablePolicy.SAVE_VARIABLE_DEVICES:
582 save_options.VariablePolicy.EXPAND_DISTRIBUTED_VARIABLES, True),
584 save_options.VariablePolicy.EXPAND_DISTRIBUTED_VARIABLES, False),
[all …]
|
| D | save_context_test.py | 21 from tensorflow.python.saved_model import save_options
31 options = save_options.SaveOptions(save_debug_info=True)
44 options = save_options.SaveOptions(save_debug_info=False)
75 options = save_options.SaveOptions()
|
| D | load_options.py | 17 from tensorflow.python.saved_model import save_options
119 save_options.VariablePolicy.from_obj(experimental_variable_policy))
|
| D | BUILD | 325 ":save_options",
344 ":save_options",
394 ":save_options",
641 name = "save_options",
642 srcs = ["save_options.py"],
653 ":save_options",
|
| D | load.py | 146 ckpt_options, save_options, filters): argument
162 self._save_options = save_options
191 if not save_options.experimental_skip_checkpoint:
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tf_function_test.py | 34 from tensorflow.python.saved_model import save_options
164 with save_context.save_context(save_options.SaveOptions()):
169 with save_context.save_context(save_options.SaveOptions()):
189 with save_context.save_context(save_options.SaveOptions()):
194 with save_context.save_context(save_options.SaveOptions()):
|
| D | tpu_replicated_variable.py | 166 def _map_resources(self, save_options): argument
169 … obj_map, resource_map = first_var._map_resources(save_options) # pylint:disable=protected-access
|
| D | ps_values.py | 227 def _map_resources(self, save_options): argument
231 obj_map, resource_map = self._v._map_resources(save_options) # pylint:disable=protected-access
516 def _map_resources(self, save_options): argument
520 obj_map, resource_map = self._v._map_resources(save_options) # pylint:disable=protected-access
|
| D | distributed_variable_test.py | 49 from tensorflow.python.saved_model import save_options
326 options = save_options.SaveOptions(
327 experimental_variable_policy=save_options.VariablePolicy.NONE)
610 options = save_options.SaveOptions()
|
| D | values_util.py | 27 from tensorflow.python.saved_model import save_options
369 save_options.VariablePolicy.EXPAND_DISTRIBUTED_VARIABLES)
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/tests/ |
| D | BUILD | 45 "//tensorflow/python/saved_model:save_options",
156 "//tensorflow/python/saved_model:save_options",
176 "//tensorflow/python/saved_model:save_options",
202 "//tensorflow/python/saved_model:save_options",
229 "//tensorflow/python/saved_model:save_options",
255 "//tensorflow/python/saved_model:save_options",
281 "//tensorflow/python/saved_model:save_options",
304 "//tensorflow/python/saved_model:save_options",
329 "//tensorflow/python/saved_model:save_options",
386 "//tensorflow/python/saved_model:save_options",
[all …]
|
| D | gen_pow_v2.py | 27 from tensorflow.python.saved_model import save_options
52 options=save_options.SaveOptions(save_debug_info=False))
|
| D | gen_data.py | 27 from tensorflow.python.saved_model import save_options
53 options=save_options.SaveOptions(save_debug_info=False))
|
| D | gen_saved_model_v2.py | 29 from tensorflow.python.saved_model import save_options
59 options=save_options.SaveOptions(save_debug_info=False))
|
| D | gen_matmul_gpu.py | 31 from tensorflow.python.saved_model import save_options
63 options=save_options.SaveOptions(save_debug_info=False))
|
| D | gen_variable_on_tpu.py | 31 from tensorflow.python.saved_model import save_options
63 options=save_options.SaveOptions(save_debug_info=False))
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/ |
| D | common.py | 79 save_options = tf.saved_model.SaveOptions(save_debug_info=show_debug_info)
81 create_module_fn(), save_model_path, options=save_options)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/debuginfo/ |
| D | saved_model_error.py | 45 save_options = tf.saved_model.SaveOptions(save_debug_info=True)
46 tf.saved_model.save(test_model, saved_model_path, options=save_options)
|
| /external/tensorflow/tensorflow/lite/examples/experimental_new_converter/ |
| D | stack_trace_example.py | 48 save_options = tf.saved_model.SaveOptions(save_debug_info=True)
49 tf.saved_model.save(test_model, saved_model_path, options=save_options)
|
| /external/tensorflow/tensorflow/cc/saved_model/testdata/ |
| D | generate_saved_models.py | 32 from tensorflow.python.saved_model import save_options
134 options = save_options.SaveOptions(save_debug_info=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.saved_model.-save-options.pbtxt | 3 is_instance: "<class \'tensorflow.python.saved_model.save_options.SaveOptions\'>"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.saved_model.-save-options.pbtxt | 3 is_instance: "<class \'tensorflow.python.saved_model.save_options.SaveOptions\'>"
|
| /external/python/cpython2/Tools/pynche/ |
| D | Switchboard.py | 113 v.save_options(self.__optiondb)
|
| /external/python/cpython3/Tools/pynche/ |
| D | Switchboard.py | 112 v.save_options(self.__optiondb)
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | autocast_variable.py | 352 def _map_resources(self, save_options): argument
355 …obj_map, resource_map = self._variable._map_resources(save_options) # pylint:disable=protected-ac…
|