| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 77 def step_fn(inputs): function
86 step_fn, args=(next(iterator),))
102 def step_fn(inputs): function
111 outputs = distribution.run(step_fn, args=(replicated_inputs,))
140 def step_fn(inputs): function
149 outputs = distribution.run(step_fn, args=(next(iterator),))
165 def step_fn(inputs): function
175 distribution.run(step_fn, args=(next(iterator),))
197 def step_fn(inputs): function
206 distribution.run(step_fn, args=(next(iterator),))
[all …]
|
| D | custom_training_loop_optimizer_test.py | 63 def step_fn(grads): function
70 distribution.run(step_fn, args=(grads,)))
90 def step_fn(grads): function
97 distribution.run(step_fn, args=(grads,)))
112 def step_fn(grads): function
117 return distribution.run(step_fn, args=(grads,))
|
| D | custom_training_loop_metrics_test.py | 50 def step_fn(): function
55 distribution.run(step_fn)
76 def step_fn(i): function
80 distribution.run(step_fn, args=(i,))
116 def step_fn(i): function
120 distribution.run(step_fn, args=(i,))
|
| D | ctl_correctness_test.py | 117 def step_fn(inputs): function
132 per_replica_losses = strategy.run(step_fn, args=(x,))
137 total_loss += step_fn(x)
143 per_replica_losses = strategy.run(step_fn, args=(next(iterator),))
148 total_loss += step_fn(next(iterator))
178 def step_fn(inputs): function
189 per_replica_losses = strategy.run(step_fn, args=(dist_inputs,))
194 return step_fn(dist_inputs)
|
| D | minimize_loss_test.py | 98 def step_fn(ctx, inputs): function
108 step_fn, iterator, iterations=2).run_op
193 def step_fn(ctx, inputs): function
203 step_fn, iterator, iterations=1).run_op
258 def step_fn(ctx, inputs): function
271 step_fn, iterator, iterations=1).run_op
368 def step_fn(ctx, inputs): function
378 step_fn, iterator, iterations=1).run_op
456 def step_fn(output_context, inputs): function
487 step_fn, iterator, iterations=2,
|
| D | mirrored_strategy_test.py | 112 def step_fn(inputs): function
125 distribution.run(step_fn, args=(next(iterator),))
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | zero_batch_test.py | 59 def step_fn(is_training, inputs, targets=None): function
74 step_fn, args=(True, inputs_placeholder, targets_placeholder))
76 step_fn, args=(False, inputs_placeholder))
130 def step_fn(inputs, targets): function
138 return distribution.run(step_fn, args=(inputs, targets))
152 def step_fn(inputs): function
156 return distribution.run(step_fn, args=(inputs,))
188 def step_fn(inputs): function
198 return distribution.run(step_fn, args=(next(iterator),))
|
| D | step_fn.py | 97 def step_fn(ctx, inputs): function
113 step_fn, self._iterator, self._iterations_per_step)
|
| D | custom_training_loop_input_test.py | 174 def step_fn(x): function
181 distribution.run(step_fn, args=(optional_data.get_value(),))
344 def step_fn(data): function
359 distribution.run(step_fn, args=(elem,)))
651 def step_fn(example): function
659 distribution.run(step_fn, args=(next(input_iterator),)))
679 def step_fn(example): function
702 distribution.run(step_fn, args=(next(input_iterator),)))
763 def step_fn(example): function
769 distribution.run(step_fn, args=(next(input_iterator),)))
[all …]
|
| D | single_loss_example.py | 22 from tensorflow.python.distribute import step_fn
47 single_loss_step = step_fn.StandardSingleLossStep(
|
| D | tpu_strategy_test.py | 297 def step_fn(inputs): function
302 strategy.run(step_fn, args=(next(iterator),)))
337 def step_fn(): function
340 all_core_strategy.run(step_fn)
341 r1 = first_core_strategy.run(step_fn)
342 r2 = second_core_strategy.run(step_fn)
387 def step_fn(): function
391 strategy.run(step_fn)
419 def step_fn(condition): function
428 return strategy.run(step_fn, args=([True, False, True],))
[all …]
|
| D | README.md | 62 def step_fn(inputs):
71 step_fn, args=(next(iterator),))
|
| D | metrics_v1_test.py | 104 def step_fn(ctx, inputs): function
111 step_fn, iterator, iterations=distribution.extended.steps_per_run)
|
| D | parameter_server_strategy_v2_test.py | 101 def step_fn(iterator): function
105 strategy.run(step_fn, args=(iter(dataset),))
|
| D | BUILD | 1436 ":step_fn",
1447 name = "step_fn",
1448 srcs = ["step_fn.py"],
|
| D | mirrored_strategy_test.py | 1388 def step_fn(): function
1392 g1, g2 = step_fn()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1052 def step_fn(step_context): function
1054 return step_fn
1085 def step_fn(step_context): function
1087 return step_fn
1118 def step_fn(step_context): function
1120 return step_fn
1164 def step_fn(step_context): function
1167 return step_fn
1200 def step_fn(step_context): function
1203 return step_fn
[all …]
|
| D | monitored_session.py | 780 def run_step_fn(self, step_fn): argument
819 step_fn_arguments = function_utils.fn_args(step_fn)
834 return self._sess.run_step_fn(step_fn, self._tf_sess(), run_with_hooks=None)
1202 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1207 return step_fn(_MonitoredSession.StepContext(raw_session, run_with_hooks))
1297 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1304 return self._sess.run_step_fn(step_fn, raw_session, run_with_hooks)
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | tpu_strategy_test.py | 108 def step_fn(i): function
116 strategy.run(step_fn, args=(i,))
168 def step_fn(inputs): function
181 strategy.run(step_fn, args=(next(iterator),))
|
| D | multi_worker_tutorial_test.py | 265 def step_fn(inputs): function
283 per_replica_losses = strategy.run(step_fn, args=(next(iterator),))
|
| /external/toolchain-utils/rust_tools/ |
| D | rust_uprev.py | 460 step_fn: Callable[[], T],
470 val = step_fn()
648 step_fn: Callable[[], T],
653 step_fn, result_from_json, result_to_json)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.-monitored-session.pbtxt | 28 argspec: "args=[\'self\', \'step_fn\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.train.-singular-monitored-session.pbtxt | 32 argspec: "args=[\'self\', \'step_fn\'], varargs=None, keywords=None, defaults=None"
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | framework.py | 643 def run_step_fn(self, step_fn): argument
644 return step_fn(
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_distributed_v1.py | 183 step_fn = _make_train_step_fn(model, ModeKeys.TRAIN, current_strategy,
194 step_fn, iterator, iterations=steps_per_run,
|