Home
last modified time | relevance | path

Searched refs:run_op (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py145 run_op = strategy.experimental_run(run_fn)
147 self._run_if_in_graph_mode(run_op)
160 run_op = get_gradients(opt, loss, [var])
164 self.evaluate(run_op)
185 run_op = strategy.experimental_run(run_fn)
187 self._run_if_in_graph_mode(run_op)
196 run_op = strategy.experimental_run(run_fn)
197 self._run_if_in_graph_mode(run_op)
216 run_op = strategy.experimental_run(run_fn)
218 self._run_if_in_graph_mode(run_op)
[all …]
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py106 run_op = strategy.experimental_run(run_fn)
108 self._run_if_in_graph_mode(run_op)
123 run_op = opt.get_gradients(loss, [var])
127 self.evaluate(run_op)
193 run_op = strategy.experimental_run(run_fn)
195 self._run_if_in_graph_mode(run_op)
203 run_op = strategy.experimental_run(run_fn)
204 self._run_if_in_graph_mode(run_op)
235 run_op = strategy.experimental_run(run_fn)
237 self._run_if_in_graph_mode(run_op)
[all …]
/external/tensorflow/tensorflow/tools/ci_build/builds/
Dtest_user_ops.sh216 function run_op() { function
243 run_op "$("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; print(tf.Session('').run(tf.load_op_lib…
247run_op "$("${PYTHON_BIN_PATH}" -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.…
/external/tensorflow/tensorflow/python/keras/distribute/
Dminimize_loss_test.py108 step_fn, iterator, iterations=2).run_op
203 step_fn, iterator, iterations=1).run_op
271 step_fn, iterator, iterations=1).run_op
378 step_fn, iterator, iterations=1).run_op
503 return (ctx.run_op, ctx.last_step_outputs["replica_loss_reduced"])
/external/tensorflow/tensorflow/python/distribute/
Dstep_fn.py114 return ctx.run_op
Dstrategy_test_lib.py587 run_op = strategy.experimental_run(comm_fn, inputs)
590 sess.run(run_op)
603 run_op = experimental_run(comm_fn, inputs)
606 sess.run(run_op)
Done_device_strategy.py380 ctx.run_op = control_flow_ops.group(loop_result)
Dmetrics_v1_test.py112 update = ctx.run_op
Dmirrored_strategy.py634 ctx.run_op = control_flow_ops.group(loop_result)
Dtpu_strategy.py1058 ctx.run_op = control_flow_ops.group(replicate_outputs)
/external/autotest/client/common_lib/hosts/
Dbase_classes.py563 self.job.run_op(op, op_func, self.get_kernel_ver)
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_test.py557 run_op = image_ops.flip_left_right(inputs)
562 self.evaluate(run_op)
587 run_op = image_ops.random_flip_left_right(inputs)
592 self.evaluate(run_op)
617 run_op = image_ops.random_flip_left_right(inputs)
622 self.evaluate(run_op)
679 run_op = control_flow_ops.group(outputs)
684 self.evaluate(run_op)
722 run_op = control_flow_ops.group(outputs)
725 self.evaluate(run_op)
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed_v1.py196 train_op = ctx.run_op
/external/autotest/server/
Dserver_job.py1056 def run_op(self, op, op_func, get_kernel_func): member in server_job