Home
last modified time | relevance | path

Searched refs:script_model (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_pdt.py193 script_model = scripted_pdt_model()
195 script_model.test_sum(
245 script_model = scripted_pdt_model()
247 script_model.test_list_to_dict(
263 script_model.test_substring(
273 script_model.test_substring(
H A Dtest_cuda.py659 script_model = torch.jit.script(model)
660 is_stream_s, a, b, c = script_model()
666 load_model = self.getExportImportCopy(script_model)
H A Dtest_freezing.py1990 script_model = torch.jit.freeze(torch.jit.script(model))
1993 self.assertEqual(model(inp), script_model(inp))
1994 FileCheck().check_not("GetAttr").run(script_model.graph)
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtests_setup.py91 script_model = torch.jit.script(model)
92 torch.jit.save(script_model, self.path)
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py528 script_model = get_script_module(eager_model, tracing, x)
530 prepare_jit(script_model, qconfig_dict)
3734 script_model = NestedModel().eval()
3736 script_model.sub1.fc.weight = torch.nn.Parameter(
3739 script_model.sub1.fc.bias = torch.nn.Parameter(
3742 script_model.sub2.fc1.weight = torch.nn.Parameter(
3745 script_model.sub2.fc1.bias = torch.nn.Parameter(
3748 script_model.sub2.fc2.weight = torch.nn.Parameter(
3751 script_model.sub2.fc2.bias = torch.nn.Parameter(
3754 script_model.fc3.weight = torch.nn.Parameter(
[all …]
/aosp_15_r20/external/pytorch/torch/jit/
H A D_recursive.py723 def script_model_defines_attr(script_model, attr): argument
724 script_attr = getattr(script_model, attr, None)
733 def add_python_attr_to_scripted_model(script_model, orig, attr): argument
734 if hasattr(orig, attr) and script_model_defines_attr(script_model, attr):
735 setattr(script_model, attr, getattr(orig, attr))
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_models_onnxruntime.py55 script_model = torch.jit.script(model)
58 script_model,
H A Donnx_test_common.py196 script_model = model if is_model_script else torch.jit.script(model)
198 script_model,
H A Dtest_utility_funs.py773 script_model = torch.jit.script(model)
780 script_model,
797 torch.onnx.export(script_model, (x, y), f, opset_version=self.opset_version)
H A Dtest_pytorch_onnx_onnxruntime.py10762 script_model = torch.jit.script(model)
10819 script_model = torch.jit.script(model)
/aosp_15_r20/external/pytorch/test/cpp/aoti_inference/
H A Dtest.cpp56 std::string script_model = "script_model_" + device + ".pt"; in test_aoti_script() local
59 STRINGIZE(CMAKE_CURRENT_BINARY_DIR)) / script_model.c_str()) in test_aoti_script()
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py836 script_model = torch.jit.script(model)
844 scripted_result = script_model(src, src_mask=src_mask)
848 script_model = torch.jit.script(model)
852 scripted_result = script_model(src, src_mask=src_mask)
874 script_model = torch.jit.script(model)
882 scripted_result = script_model(src, mask=src_mask)
886 script_model = torch.jit.script(model)
890 scripted_result = script_model(src, mask=src_mask)
/aosp_15_r20/external/pytorch/docs/source/
H A Dpackage.rst471 e.save_pickle("res", "script_model.pkl", scripted_model)
475 loaded_script = importer.load_pickle("res", "script_model.pkl")