| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | monitored_session_test.py | 1089 def step_fn(step_context): function
1122 def step_fn(step_context): function
1155 def step_fn(step_context): function
1201 def step_fn(step_context): function
1237 def step_fn(step_context): function
1272 def step_fn(step_context): function
1953 def step_fn(step_context): function
1963 def step_fn(step_context): function
1974 def step_fn(step_context): function
1988 def step_fn(step_context): function
[all …]
|
| H A D | monitored_session.py | 784 def run_step_fn(self, step_fn): argument
1230 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
1325 def run_step_fn(self, step_fn, raw_session, run_with_hooks): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | zero_batch_test.py | 56 def step_fn(is_training, inputs, targets=None): function
127 def step_fn(inputs, targets): function
149 def step_fn(inputs): function
185 def step_fn(inputs): function
|
| H A D | custom_training_loop_input_test.py | 172 def step_fn(x): function
342 def step_fn(data): function
676 def step_fn(example): function
704 def step_fn(example): function
791 def step_fn(example): function
910 def step_fn(data): function
|
| H A D | tpu_strategy_test.py | 303 def step_fn(inputs): function
434 def step_fn(inputs): function
474 def step_fn(): function
524 def step_fn(): function
556 def step_fn(condition): function
576 def step_fn(prev): function
1119 def step_fn(inputs): function
|
| H A D | tpu_strategy_model_parallelism_test.py | 207 def step_fn(batch_features): function
339 def step_fn(x): function
|
| H A D | parameter_server_strategy_v2_test.py | 190 def step_fn(iterator): function
204 def step_fn(iterator): function
|
| H A D | step_fn.py | 93 def step_fn(ctx, inputs): function
|
| H A D | metrics_v1_test.py | 99 def step_fn(ctx, inputs): function
|
| H A D | mirrored_strategy_test.py | 1418 def step_fn(): function
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_memory_profiler.py | 1115 def step_fn(_): function
1134 def step_fn(mark_region): function
1182 def step_fn(mark_region): function
1246 def step_fn(mark_region): function
1282 def step_fn(mark_region): function
1357 def step_fn(mark_region): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/ |
| H A D | tpu_memory_test.py | 145 def step_fn(inputs): function
|
| /aosp_15_r20/external/grpc-grpc/test/core/end2end/ |
| H A D | cq_verifier.cc | 202 CqVerifier::CqVerifier( in CqVerifier()
|
| H A D | end2end_tests.h | 164 void SetCqVerifierStepFn( in SetCqVerifierStepFn()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/debug/wrappers/ |
| H A D | framework.py | 636 def run_step_fn(self, step_fn): argument
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | lr_scheduler.py | 129 def wrap_step(step_fn): argument
|