Searched refs:experimental_run_v2 (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | strategy_test_lib.py | 386 out1 = strategy.experimental_run_v2(lambda: constant_op.constant(4.))
389 out2 = strategy.experimental_run_v2(
395 out3 = strategy.experimental_run_v2(lambda b, a: a + 2 * b + 2, kwargs=out2)
475 out1 = strategy.experimental_run_v2(
479 out2 = strategy.experimental_run_v2(
485 out3 = strategy.experimental_run_v2(lambda b, a: a + 2 * b + 2, kwargs=out2)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | template_mirrored_strategy_test.py | 45 out = strategy.unwrap(strategy.experimental_run_v2(temp))
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy.pbtxt | 38 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 43 name: "experimental_run_v2"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy.pbtxt | 38 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 43 name: "experimental_run_v2"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 39 name: "experimental_run_v2"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 471 return self.experimental_run_v2(fn, args=args)
473 def experimental_run_v2(self, fn, args=(), kwargs=None): member in DistributionStrategy
|
| D | tpu_strategy.py | 157 def experimental_run_v2(self, fn, args=(), kwargs=None): member in TPUStrategy
|