| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 77 def step_fn(inputs): function
102 def step_fn(inputs): function
140 def step_fn(inputs): function
165 def step_fn(inputs): function
197 def step_fn(inputs): function
241 def step_fn(inputs): function
295 def step_fn(inputs): function
312 def step_fn(inputs): function
350 def step_fn(inputs): function
377 def step_fn(inputs): function
[all …]
|
| D | custom_training_loop_metrics_test.py | 50 def step_fn(): function
76 def step_fn(i): function
116 def step_fn(i): function
|
| D | custom_training_loop_optimizer_test.py | 63 def step_fn(grads): function
90 def step_fn(grads): function
112 def step_fn(grads): function
|
| D | minimize_loss_test.py | 98 def step_fn(ctx, inputs): function
193 def step_fn(ctx, inputs): function
258 def step_fn(ctx, inputs): function
368 def step_fn(ctx, inputs): function
456 def step_fn(output_context, inputs): function
|
| D | ctl_correctness_test.py | 117 def step_fn(inputs): function
178 def step_fn(inputs): function
|
| D | mirrored_strategy_test.py | 112 def step_fn(inputs): function
|
| D | distribute_strategy_test.py | 2393 def step_fn(inputs): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | zero_batch_test.py | 59 def step_fn(is_training, inputs, targets=None): function
130 def step_fn(inputs, targets): function
152 def step_fn(inputs): function
188 def step_fn(inputs): function
|
| D | custom_training_loop_input_test.py | 174 def step_fn(x): function
344 def step_fn(data): function
651 def step_fn(example): function
679 def step_fn(example): function
763 def step_fn(example): function
882 def step_fn(data): function
|
| D | step_fn.py | 97 def step_fn(ctx, inputs): function
|
| D | tpu_strategy_test.py | 297 def step_fn(inputs): function
337 def step_fn(): function
387 def step_fn(): function
419 def step_fn(condition): function
439 def step_fn(prev): function
|
| D | parameter_server_strategy_v2_test.py | 101 def step_fn(iterator): function
|
| D | metrics_v1_test.py | 104 def step_fn(ctx, inputs): function
|
| D | mirrored_strategy_test.py | 1388 def step_fn(): function
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1052 def step_fn(step_context): function
1085 def step_fn(step_context): function
1118 def step_fn(step_context): function
1164 def step_fn(step_context): function
1200 def step_fn(step_context): function
1235 def step_fn(step_context): function
1916 def step_fn(step_context): function
1926 def step_fn(step_context): function
1937 def step_fn(step_context): function
1951 def step_fn(step_context): function
[all …]
|
| D | monitored_session.py | 780 def run_step_fn(self, step_fn): argument
1202 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1297 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | tpu_strategy_test.py | 108 def step_fn(i): function
168 def step_fn(inputs): function
|
| D | multi_worker_tutorial_test.py | 265 def step_fn(inputs): function
|
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| D | framework.py | 643 def run_step_fn(self, step_fn): argument
|