Home
last modified time | relevance | path

Searched refs:OneDeviceStrategy (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py40 class OneDeviceStrategy(distribute_lib.Strategy): class
79 super(OneDeviceStrategy, self).__init__(OneDeviceExtended(self, device))
109 return super(OneDeviceStrategy, self).experimental_distribute_dataset(
153 return super(OneDeviceStrategy,
170 return super(OneDeviceStrategy, self).experimental_local_results(value)
188 return super(OneDeviceStrategy, self).run(fn, args, kwargs, options)
221 return super(OneDeviceStrategy, self).reduce(reduce_op, value, axis)
237 return super(OneDeviceStrategy, self).scope()
243 __doc__ = OneDeviceStrategy.__doc__.replace(
251 __init__.__doc__ = OneDeviceStrategy.__init__.__doc__
Dstrategy_combinations.py64 OneDeviceStrategy = _version_chooser(one_device_lib.OneDeviceStrategyV1, variable
65 one_device_lib.OneDeviceStrategy)
220 "OneDeviceCPU", lambda: OneDeviceStrategy("/cpu:0"), required_gpus=None)
222 "OneDeviceGPU", lambda: OneDeviceStrategy("/gpu:0"), required_gpus=1)
225 lambda: OneDeviceStrategy("/job:worker/replica:0/task:1/cpu:0"),
229 lambda: OneDeviceStrategy("/job:worker/replica:0/task:1/gpu:0"),
Dstrategy_combinations_test.py155 self.assertIsInstance(strategy, one_device_strategy.OneDeviceStrategy)
/external/tensorflow/tensorflow/python/keras/benchmarks/
Ddistribution_util.py122 return tf.distribute.OneDeviceStrategy("device:CPU:0")
126 return tf.distribute.OneDeviceStrategy("device:GPU:0")
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-one-device-strategy.pbtxt1 path: "tensorflow.distribute.OneDeviceStrategy"
3 is_instance: "<class \'tensorflow.python.distribute.one_device_strategy.OneDeviceStrategy\'>"
Dtensorflow.distribute.pbtxt48 name: "OneDeviceStrategy"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.pbtxt28 name: "OneDeviceStrategy"
Dtensorflow.distribute.-one-device-strategy.pbtxt1 path: "tensorflow.distribute.OneDeviceStrategy"
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py1185 one_device_strategy.OneDeviceStrategy,
/external/tensorflow/tensorflow/python/keras/engine/
Dcompile_utils_test.py737 distribution = one_device_strategy.OneDeviceStrategy('/device:CPU:0')
/external/tensorflow/tensorflow/python/autograph/g3doc/reference/
Dcontrol_flow.md323 for i in tf.distribute.OneDeviceStrategy('cpu').experimental_distribute_dataset(
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dtext_vectorization_test.py787 strategy = one_device_strategy.OneDeviceStrategy("/cpu:0")