| /external/pytorch/test/inductor/ |
| D | test_aot_inductor_package.py | 39 def check_model( function
85 self.check_model(Model(), example_inputs)
100 self.check_model(Model(), example_inputs)
109 check_model = check_model variable in AOTInductorTestPackagedABICompatibleCuda
122 check_model = check_model variable in AOTInductorTestPackagedABICompatibleCpu
|
| D | test_aot_inductor.py | 93 def check_model( function
188 self.check_model(Model(), example_inputs)
201 self.check_model(Model().to(self.device), example_inputs)
217 self.check_model(Model(), example_inputs)
254 self.check_model(Model(self.device), example_inputs)
272 self.check_model(Model(self.device), example_inputs)
285 self.check_model(Model(), example_inputs)
320 self.check_model(Model(), example_inputs)
340 self.check_model(Model(self.device), example_inputs)
363 self.check_model(Model(self.device), example_inputs)
[all …]
|
| D | test_torchinductor_opinfo.py | 50 from .test_torchinductor import check_model, check_model_gpu
52 from test_torchinductor import check_model, check_model_gpu
563 check_model = check_model variable in TestInductorOpInfo
772 self.check_model(
|
| D | test_combo_kernels.py | 20 from .test_torchinductor import check_model, check_model_cuda
22 from test_torchinductor import check_model, check_model_cuda
33 check_model_cpu = check_model
154 check_model_cpu = check_model
300 check_model_cpu = check_model
|
| D | test_binary_folding.py | 19 from inductor.test_torchinductor import check_model, check_model_gpu, copy_tests
233 common = check_model
|
| D | test_benchmark_fusion.py | 22 from inductor.test_torchinductor import check_model, check_model_cuda, copy_tests
297 common = check_model
|
| D | test_compiled_optimizers.py | 271 from .test_torchinductor import check_model, check_model_gpu
273 from test_torchinductor import check_model, check_model_gpu
601 check_model_cpu = check_model
|
| D | test_foreach.py | 22 from .test_torchinductor import check_model, check_model_cuda
24 from test_torchinductor import check_model, check_model_cuda
90 check_model_cpu = check_model
|
| D | test_cpu_select_algorithm.py | 49 check_model = test_torchinductor.check_model variable
143 common = check_model
1458 common = check_model
|
| D | test_torchinductor_dynamic_shapes.py | 43 check_model,
96 common = check_model
|
| D | test_inductor_freezing.py | 26 from inductor.test_torchinductor import check_model, check_model_cuda, copy_tests
949 common = check_model
|
| D | test_cpu_repro.py | 66 check_model = test_torchinductor.check_model variable
113 common = check_model
|
| D | test_torchinductor.py | 366 def check_model( function
599 check_model(
631 check_model(
728 check_model(
10540 common = check_model
|
| /external/pytorch/test/onnx/ |
| D | test_fx_to_onnx.py | 395 onnx.checker.check_model(onnx_program.model_proto, full_check=True)
508 onnx.checker.check_model(onnx_program.model_proto)
530 onnx.checker.check_model(onnx_program.model_proto)
651 onnx.checker.check_model(tmp_onnx_file.name, full_check=True)
|
| D | test_onnx_opset.py | 29 onnx.checker.check_model(model)
|
| D | test_pytorch_onnx_no_runtime.py | 73 onnx.checker.check_model(onnx_model)
1112 onnx.checker.check_model(f.getvalue())
|
| D | test_operators.py | 101 onnx.checker.check_model(model_def)
367 onnx.checker.check_model(onnx_model)
|
| /external/pytorch/torch/onnx/_internal/exporter/ |
| D | _core.py | 1225 onnxscript_apis.check_model(onnx_program.model)
|
| /external/pytorch/torch/csrc/jit/serialization/ |
| D | export.cpp | 1454 onnx::checker::check_model(model); in check_onnx_proto()
|
| /external/pytorch/docs/source/ |
| D | onnx_torchscript.rst | 84 onnx.checker.check_model(model)
|