Searched refs:experimental_run (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | strategy_test_lib.py | 367 x, y = strategy.experimental_run(lambda z: z, i)
429 strategy.experimental_run(comm_fn, inputs))))
450 strategy.experimental_run(step, inputs))))
468 strategy.experimental_run(step, inputs))))
523 strategy.experimental_run(comm_fn, inputs))))
544 strategy.experimental_run(step, inputs))))
562 strategy.experimental_run(step, inputs))))
|
| /external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
| D | mnist_eager_multigpu.py | 136 dist_train = lambda it: strategy.experimental_run(train_step, it)
137 dist_test = lambda it: strategy.experimental_run(test_step, it)
|
| D | mnist_tf1_tpu.py | 133 strategy.experimental_run(train_step, train_iterator))
135 strategy.experimental_run(test_step, test_iterator))
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/ |
| D | loss_scale_optimizer_test.py | 85 run_op = strategy.experimental_run(run_fn)
|
| D | keras_test.py | 239 op = strategy.experimental_run(run_fn)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy.pbtxt | 34 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 39 name: "experimental_run"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 35 name: "experimental_run"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy.pbtxt | 34 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 39 name: "experimental_run"
|
| D | tensorflow.distribute.-mirrored-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.experimental.-parameter-server-strategy.pbtxt | 35 name: "experimental_run"
|
| D | tensorflow.distribute.-one-device-strategy.pbtxt | 35 name: "experimental_run"
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 440 def experimental_run(self, fn, input_iterator=None): member in DistributionStrategy
|