| /aosp_15_r20/external/pytorch/android/pytorch_android/ |
| H A D | generate_test_torchscripts.py | 22 @torch.jit.script_method
26 @torch.jit.script_method
30 @torch.jit.script_method
34 @torch.jit.script_method
38 @torch.jit.script_method
42 @torch.jit.script_method
46 @torch.jit.script_method
50 @torch.jit.script_method
54 @torch.jit.script_method
58 @torch.jit.script_method
[all …]
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | android_api_module.py | 8 @torch.jit.script_method
12 @torch.jit.script_method
16 @torch.jit.script_method
20 @torch.jit.script_method
24 @torch.jit.script_method
28 @torch.jit.script_method
32 @torch.jit.script_method
36 @torch.jit.script_method
40 @torch.jit.script_method
44 @torch.jit.script_method
[all …]
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | mkldnn.py | 19 @torch.jit.script_method
23 @torch.jit.script_method
29 @torch.jit.script_method
60 @torch.jit.script_method
64 @torch.jit.script_method
82 @torch.jit.script_method
100 @torch.jit.script_method
122 @torch.jit.script_method
155 @torch.jit.script_method
163 @torch.jit.script_method
[all …]
|
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | custom_lstms.py | 132 @jit.script_method
170 @jit.script_method
176 @jit.script_method
200 @jit.script_method
226 @jit.script_method
243 @jit.script_method
267 @jit.script_method
301 @jit.script_method
331 @jit.script_method
371 @jit.script_method
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 485 @torch.jit.script_method
955 @torch.jit.script_method
991 @torch.jit.script_method
1016 @torch.jit.script_method
2457 @torch.jit.script_method
2466 @torch.jit.script_method
2653 @torch.jit.script_method
2672 @torch.jit.script_method
2749 @torch.jit.script_method
2820 @torch.jit.script_method
[all …]
|
| H A D | test_throughput_benchmark.py | 14 @torch.jit.script_method
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_no_runtime.py | 95 @torch.jit.script_method
122 @torch.jit.script_method
156 @torch.jit.script_method
176 @torch.jit.script_method
186 @torch.jit.script_method
195 @torch.jit.script_method
205 @torch.jit.script_method
220 @torch.jit.script_method
233 @torch.jit.script_method
243 @torch.jit.script_method
[all …]
|
| H A D | test_pytorch_onnx_onnxruntime.py | 463 @torch.jit.script_method
935 @torch.jit.script_method
965 @torch.jit.script_method
976 @torch.jit.script_method
987 @torch.jit.script_method
995 @torch.jit.script_method
1006 @torch.jit.script_method
1020 @torch.jit.script_method
1041 @torch.jit.script_method
1127 @torch.jit.script_method
[all …]
|
| H A D | test_onnx_opset.py | 110 @torch.jit.script_method
244 @torch.jit.script_method
|
| H A D | test_onnxscript_no_runtime.py | 117 @torch.jit.script_method
|
| /aosp_15_r20/external/pytorch/test/mobile/ |
| H A D | test_lite_script_module.py | 333 @torch.jit.script_method
348 @torch.jit.script_method
368 @torch.jit.script_method
372 @torch.jit.script_method
396 @torch.jit.script_method
415 @torch.jit.script_method
421 @torch.jit.script_method
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_logging.py | 26 @torch.jit.script_method
86 @torch.jit.script_method
|
| H A D | test_data_parallel.py | 67 @torch.jit.script_method
76 @torch.jit.script_method
|
| H A D | test_module_interface.py | 413 @torch.jit.script_method
417 @torch.jit.script_method
422 @torch.jit.script_method
426 @torch.jit.script_method
|
| H A D | test_async.py | 96 @torch.jit.script_method
100 @torch.jit.script_method
261 @torch.jit.script_method
|
| H A D | test_complex.py | 49 @torch.jit.script_method
201 @torch.jit.script_method
|
| H A D | test_tracer.py | 2367 @torch.jit.script_method
2376 @torch.jit.script_method
2380 @torch.jit.script_method
2424 @torch.jit.script_method
2441 @torch.jit.script_method
2476 @torch.jit.script_method
2536 @torch.jit.script_method
2561 @torch.jit.script_method
2578 @torch.jit.script_method
|
| H A D | test_models.py | 503 @torch.jit.script_method
649 @torch.jit.script_method
712 @torch.jit.script_method
|
| H A D | test_class_type.py | 249 @torch.jit.script_method
279 @torch.jit.script_method
318 @torch.jit.script_method
555 @torch.jit.script_method
|
| H A D | test_exception.py | 19 @torch.jit.script_method
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _script.py | 348 def script_method(fn): function
820 script_method = self._c._get_method(attr)
823 self.__dict__[attr] = script_method
824 return script_method
|
| H A D | __init__.py | 43 script_method,
|
| /aosp_15_r20/external/pytorch/tools/coverage_plugins_package/ |
| H A D | README.md | 3 …unctions and methods passed through `torch.jit.script` and `torch.jit.script_method` as covered co…
|
| /aosp_15_r20/external/pytorch/test/custom_operator/ |
| H A D | model.py | 25 @torch.jit.script_method
|
| /aosp_15_r20/external/pytorch/test/cpp/jit/ |
| H A D | tests_setup.py | 32 @torch.jit.script_method
|