Home
last modified time | relevance | path

Searched refs:experimental_run (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py145 run_op = strategy.experimental_run(run_fn)
185 run_op = strategy.experimental_run(run_fn)
196 run_op = strategy.experimental_run(run_fn)
216 run_op = strategy.experimental_run(run_fn)
227 run_op = strategy.experimental_run(run_fn)
250 run_op = strategy.experimental_run(run_fn)
260 run_op = strategy.experimental_run(run_fn)
288 opt_op = strategy.experimental_run(run_fn)
300 self.evaluate(strategy.experimental_run(run_fn))
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py106 run_op = strategy.experimental_run(run_fn)
193 run_op = strategy.experimental_run(run_fn)
203 run_op = strategy.experimental_run(run_fn)
235 run_op = strategy.experimental_run(run_fn)
245 run_op = strategy.experimental_run(run_fn)
255 run_op = strategy.experimental_run(run_fn)
272 run_op = strategy.experimental_run(run_fn)
283 run_op = strategy.experimental_run(run_fn)
303 run_op = strategy.experimental_run(run_fn)
326 run_op = strategy.experimental_run(run_fn)
[all …]
Dkeras_test.py271 op = strategy.experimental_run(run_fn)
/external/tensorflow/tensorflow/python/distribute/
Dstrategy_test_lib.py455 x, y = strategy.experimental_run(
535 strategy.experimental_run(comm_fn, inputs))))
557 strategy.experimental_run(step, inputs))))
577 strategy.experimental_run(step, inputs))))
587 run_op = strategy.experimental_run(comm_fn, inputs)
600 experimental_run = def_function.function()(strategy.experimental_run)
603 run_op = experimental_run(comm_fn, inputs)
674 strategy.experimental_run(
698 strategy.experimental_run(
720 strategy.experimental_run(
Ddistribute_lib.py942 def experimental_run(self, fn, input_iterator=None): member in StrategyBase
1919 def experimental_run(self, fn, input_iterator=None): # pylint: disable=useless-super-delegation member in StrategyV1
1951 return super(StrategyV1, self).experimental_run(
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-mirrored-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt53 name: "experimental_run"
Dtensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.-one-device-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.-strategy.pbtxt51 name: "experimental_run"
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.-t-p-u-strategy.pbtxt60 name: "experimental_run"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-strategy.pbtxt51 name: "experimental_run"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.-mirrored-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.-one-device-strategy.pbtxt52 name: "experimental_run"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt56 name: "experimental_run"
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt52 name: "experimental_run"