Searched refs:distributed_values (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_gather_test.py | 74 distributed_values = strategy.experimental_distribute_values_from_function(
78 return strategy.gather(distributed_values, axis=axis)
132 distributed_values = strategy.experimental_distribute_values_from_function(
137 return strategy.gather(distributed_values, axis=axis)
154 distributed_values = strategy.experimental_distribute_values_from_function(
159 return strategy.gather(distributed_values, axis=axis)
182 distributed_values = strategy.experimental_distribute_values_from_function(
187 return strategy.gather(distributed_values, axis=axis)
234 distributed_values = strategy.experimental_distribute_values_from_function(
239 return strategy.gather(distributed_values, axis=axis)
|
| D | values_test.py | 131 distributed_values = (
134 ds_test_util.gather(distribution, distributed_values),
151 distributed_values = (
154 ds_test_util.gather(distribution, distributed_values).numpy(),
170 distributed_values = (
172 distributed_values = ds_test_util.gather(distribution, distributed_values)
178 self.assertAllEqual(distributed_values, expected)
195 distributed_values = (
197 distributed_values = ds_test_util.gather(distribution, distributed_values)
203 self.assertAllEqual(distributed_values, expected)
[all …]
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 214 distributed_values = strategy.run(run)
216 tf.distribute.ReduceOp.SUM, distributed_values, axis=None)
218 values = strategy.experimental_local_results(distributed_values)
|
| /external/tensorflow/tensorflow/python/module/ |
| D | module_test.py | 31 from tensorflow.python.distribute import values as distributed_values unknown
252 mirrored = distributed_values.MirroredVariable(
|