| /external/pytorch/benchmarks/dynamo/ |
| 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 …]
|
| /external/pytorch/torch/_dynamo/backends/ |
| 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
|
| 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
|
| /external/executorch/backends/xnnpack/test/ |
| 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
|
| /external/pytorch/torch/_dynamo/repro/ |
| 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
|
| /external/pytorch/test/inductor/ |
| 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
|
| D | test_aot_inductor_package.py | 26 def compile(model, example_inputs, dynamic_shapes, options, device): argument
42 example_inputs, argument
|
| D | test_aot_inductor_utils.py | 29 example_inputs, argument
118 example_inputs, argument
|
| 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
|
| /external/pytorch/test/dynamo/ |
| 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
|
| 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
|
| /external/executorch/exir/tests/ |
| 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
|
| /external/executorch/examples/apple/coreml/scripts/ |
| 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
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| 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
|
| /external/executorch/examples/apple/mps/scripts/ |
| 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
|
| /external/executorch/examples/xnnpack/quantization/ |
| D | utils.py | 16 def quantize(model, example_inputs): argument
|
| D | example.py | 45 def verify_xnnpack_quantizer_matching_fx_quant_model(model_name, model, example_inputs): argument
|
| /external/executorch/examples/cadence/models/ |
| D | resnet50.py | 28 example_inputs = (torch.randn(1, 3, 64, 64),) variable
|
| D | mobilenet_v2.py | 28 example_inputs = (torch.randn(1, 3, 64, 64),) variable
|
| D | vision_transformer.py | 26 example_inputs = (torch.randn(1, 3, 224, 224),) variable
|
| D | rnnt_joiner.py | 60 example_inputs = ( variable
|
| /external/pytorch/benchmarks/dynamo/microbenchmarks/ |
| D | microbench.py | 26 def compute_speedups(args, models, example_inputs): argument
41 def microbenchmark(args, model, example_inputs): argument
|
| /external/executorch/backends/xnnpack/test/models/ |
| D | mobilebert.py | 17 example_inputs = (torch.tensor([[101, 7592, 1010, 2026, 3899, 2003, 10140, 102]]),) variable in TestMobilebert
|
| /external/executorch/examples/cadence/operators/ |
| D | quantized_conv1d_op.py | 56 example_inputs = (torch.randn(shape),) variable
|
| /external/executorch/examples/llm_manual/ |
| D | export_nanogpt.py | 22 example_inputs = ( variable
|