Searched refs:experimental_io_device (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | checkpoint_options.py | 45 def __init__(self, experimental_io_device=None): argument
59 self.experimental_io_device = experimental_io_device
|
| D | functional_saver.py | 81 save_device = options.experimental_io_device or "cpu:0"
106 restore_device = options.experimental_io_device or "cpu:0"
280 options.experimental_io_device or
|
| D | functional_saver_test.py | 54 experimental_io_device=LOCALHOST)
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_options.py | 36 experimental_io_device=None): argument
57 self.experimental_io_device = experimental_io_device
|
| D | save_options.py | 110 experimental_io_device=None, argument
164 self.experimental_io_device = experimental_io_device
|
| D | load.py | 891 experimental_io_device=options.experimental_io_device)
|
| D | save_test.py | 810 self.assertIsNone(options.experimental_io_device)
811 options = save_options.SaveOptions(experimental_io_device="/job:localhost")
812 self.assertEqual("/job:localhost", options.experimental_io_device)
|
| D | save.py | 1094 experimental_io_device=options.experimental_io_device)
|
| D | load_test.py | 1946 self.assertIsNone(options.experimental_io_device)
1947 options = load_options.LoadOptions(experimental_io_device="/job:localhost")
1948 self.assertEqual("/job:localhost", options.experimental_io_device)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.saved_model.-load-options.pbtxt | 6 name: "experimental_io_device"
11 …argspec: "args=[\'self\', \'experimental_io_device\'], varargs=None, keywords=None, defaults=[\'No…
|
| D | tensorflow.train.-checkpoint-options.pbtxt | 6 name: "experimental_io_device"
11 …argspec: "args=[\'self\', \'experimental_io_device\'], varargs=None, keywords=None, defaults=[\'No…
|
| D | tensorflow.saved_model.-save-options.pbtxt | 6 name: "experimental_io_device"
27 …espace_whitelist\', \'save_debug_info\', \'function_aliases\', \'experimental_io_device\', \'exper…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-checkpoint-options.pbtxt | 6 name: "experimental_io_device"
11 …argspec: "args=[\'self\', \'experimental_io_device\'], varargs=None, keywords=None, defaults=[\'No…
|
| D | tensorflow.saved_model.-save-options.pbtxt | 6 name: "experimental_io_device"
27 …espace_whitelist\', \'save_debug_info\', \'function_aliases\', \'experimental_io_device\', \'exper…
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | saved_model_save_load_test.py | 173 experimental_io_device='/job:localhost')
176 experimental_io_device='/job:localhost')
|
| D | checkpointing_test.py | 118 experimental_io_device="/job:localhost")
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | base.py | 351 io_device = self._checkpoint.options.experimental_io_device or "cpu:0"
|
| D | util_test.py | 265 experimental_io_device=localhost)
|
| /external/tensorflow/ |
| D | RELEASE.md | 1103 …* Add `tf.saved_model.LoadOptions` with [`experimental_io_device`](https://www.tensorflow.org/vers…
1104 …* Update `tf.saved_model.SaveOptions` with [`experimental_io_device`](https://www.tensorflow.org/v…
|