| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer_test.py | 210 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
221 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
244 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
282 run_fn = lambda: opt.minimize(lambda: var + 1., var_list=[var]) function
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | fault_tolerance_test.py | 336 def run_fn(): function
370 def run_fn(): function
409 def run_fn(): function
|
| /external/pytorch/test/ |
| D | test_utils.py | 282 def run_fn(input): function
310 def run_fn(input): function
336 def run_fn(input): function
344 def run_fn(tensor1, tensor2): function
415 def run_fn(tensor1, tensor2): function
|
| D | test_autograd.py | 5407 def run_fn(a): function
|
| /external/pytorch/torch/ao/quantization/ |
| D | quantize_jit.py | 221 run_fn=None, argument
248 def quantize_jit(model, qconfig_dict, run_fn, run_args, inplace=False, debug=False): argument
|
| D | quantize.py | 449 def quantize(model, run_fn, run_args, mapping=None, inplace=False): argument
597 def quantize_qat(model, run_fn, run_args, inplace=False): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | layer_test.py | 214 def run_fn(): function
|
| /external/grpc-grpc/test/core/event_engine/test_suite/tests/ |
| D | timer_test.cc | 210 absl::FunctionRef<void(absl::AnyInvocable<void()>)> run_fn, in ImmediateRunTestInternal()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 157 def run_fn(): function
|
| D | strategy_test_lib.py | 272 def run_fn(): function
|
| D | tpu_strategy.py | 1056 def run_fn(inputs): function
|
| D | mirrored_strategy_test.py | 96 def run_fn(): function
|
| /external/pytorch/test/dynamo/ |
| D | test_autograd_function.py | 759 def run_fn(a): function
|
| D | test_misc.py | 7426 def run_fn(): function
|