| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | common.py | 466 example_inputs, argument
510 def _normalize_bench_inputs(example_inputs) -> Tuple[Tuple[Any], Mapping[str, Any]]: argument
561 def coverage_experiment(args, model_iter_fn, model, example_inputs): argument
597 def speedup_experiment_fx2trt(args, model_iter_fn, model, example_inputs): argument
606 def recompile_profiler_experiment(args, model_iter_fn, model, example_inputs): argument
649 def speedup_experiment(args, model_iter_fn, model, example_inputs, **kwargs): argument
807 def speedup_experiment_ds(args, model_iter_fn, model, example_inputs): argument
904 example_inputs, argument
1056 def print_fx(gm, example_inputs): argument
1061 def print_aten_ops(gm, example_inputs): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | debugging.py | 91 def boxed_nop(fx_g, example_inputs): argument
165 def relu_compile_error_TESTING_ONLY(gm: torch.fx.GraphModule, example_inputs): argument
173 def relu_runtime_error_TESTING_ONLY(gm: torch.fx.GraphModule, example_inputs): argument
183 def relu_accuracy_error_TESTING_ONLY(gm: torch.fx.GraphModule, example_inputs): argument
194 def non_leaf_compile_error_TESTING_ONLY(gm: torch.fx.GraphModule, example_inputs): argument
320 def __call__(self, gm: torch.fx.GraphModule, example_inputs): argument
|
| H A D | common.py | 23 def __call__(self, gm: torch.fx.GraphModule, example_inputs, **kwargs): argument
119 def device_from_inputs(example_inputs) -> torch.device: argument
125 def dtype_from_inputs(example_inputs) -> torch.dtype: argument
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ |
| H A D | test_xnnpack_utils.py | 255 example_inputs, argument
277 example_inputs, argument
315 example_inputs, argument
469 def _get_dqlinear_graph_module(self, weight_qconfig, linear, example_inputs): argument
520 self, LinearModule, example_inputs argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/repro/ |
| H A D | after_dynamo.py | 50 def _accuracy_fails(gm, example_inputs, compiler_fn): argument
72 def __call__(self, gm, example_inputs, **kwargs): argument
275 def dynamo_minifier_backend(gm, example_inputs, compiler_name): argument
315 def dynamo_accuracy_minifier_backend(gm, example_inputs, compiler_name): argument
345 def backend_fails(gm, example_inputs, compiler_fn, orig_failure): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_backends.py | 52 def compiler_fn(graph, example_inputs): argument
80 def compiler_fn(graph, example_inputs): argument
237 def my_custom_backend(gm, example_inputs): argument
255 def my_compiler(gm, example_inputs): argument
275 def my_compiler(gm, example_inputs): argument
|
| H A D | test_verify_correctness.py | 68 def compiler_fn(graph, example_inputs): argument
108 def incorrect_compile_fn(gm, example_inputs): argument
131 def incorrect_compile_fn(gm, example_inputs): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_codegen_dynamic_shapes.py | 40 example_inputs, argument
398 def common(self: TestCase, model, example_inputs, kwargs=None, **_rest): argument
421 def common(self: TestCase, model, example_inputs, kwargs=None, **_rest): argument
|
| H A D | test_aot_inductor_package.py | 26 def compile(model, example_inputs, dynamic_shapes, options, device): argument
42 example_inputs, argument
|
| H A D | test_aot_inductor_utils.py | 29 example_inputs, argument
118 example_inputs, argument
|
| H A D | test_binary_folding.py | 76 def my_inner_compile(gm, example_inputs, *args, **kwargs): argument
186 def my_inner_compile(gm, example_inputs, *args, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_inductor_collectives.py | 75 def compile(func, example_inputs): argument
110 def compile(func, example_inputs): argument
188 def compile(func, example_inputs): argument
227 def compile(func, example_inputs): argument
353 def compile(func, example_inputs): argument
380 def compile(func, example_inputs): argument
|
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | autograd_cache.py | 166 example_inputs, argument
230 example_inputs, argument
255 def load(self, example_inputs, fx_config: Dict[str, BoxedBool]) -> CompiledFxGraph: argument
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_quantize_io_pass.py | 37 def _quantize(self, mod, example_inputs): argument
54 def _get_edge_prog_manager(self, mod, example_inputs): argument
|
| /aosp_15_r20/external/executorch/examples/apple/coreml/scripts/ |
| H A D | export.py | 89 def lower_module_to_coreml(module, compile_specs, example_inputs): argument
107 def export_lowered_module_to_executorch_program(lowered_module, example_inputs): argument
|
| /aosp_15_r20/external/executorch/examples/apple/mps/scripts/ |
| H A D | mps_example.py | 40 def get_bundled_program(executorch_program, example_inputs, expected_output): argument
155 model, example_inputs, _, _ = EagerModelFactory.create_model(**model_config) variable
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/ |
| H A D | microbench.py | 26 def compute_speedups(args, models, example_inputs): argument
41 def microbenchmark(args, model, example_inputs): argument
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | debug_utils.py | 296 def clone_inputs_retaining_gradness(example_inputs): argument
339 example_inputs, argument
439 example_inputs, argument
|
| H A D | utils.py | 864 def clone_inputs(example_inputs): argument
926 def torchscript(model, example_inputs, verbose=False): argument
1016 def timed(model, example_inputs, times=1): argument
1033 def check_is_cuda(gm, example_inputs): argument
1604 def __call__(self, gm: torch.fx.GraphModule, example_inputs): argument
|
| H A D | testing.py | 53 def collect_results(model, prediction, loss, example_inputs): argument
273 def dummy_fx_compile(gm: fx.GraphModule, example_inputs): argument
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_utils.py | 295 example_inputs, argument
317 example_inputs, argument
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _trace.py | 611 def make_tuple(example_inputs): argument
672 example_inputs=None, argument
829 example_inputs=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/backends/xnnpack/ |
| H A D | xnnpack_graph_builder.cpp | 26 std::vector<c10::IValue>& example_inputs) { in optimizeAndTraceGraph()
39 std::vector<c10::IValue> example_inputs) { in buildXNNGraph()
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_metadata_porting.py | 62 self, model, example_inputs, from_node_to_tags argument
94 example_inputs, argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | pre_grad.py | 111 def pre_grad_passes(gm: torch.fx.GraphModule, example_inputs=None): argument
291 def fuse_fx(gm: torch.fx.GraphModule, example_inputs) -> torch.fx.GraphModule: argument
|