Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/onnx/dynamo/
H A Dtest_dynamo_with_onnxruntime_backend.py160 compiled_model = torch.compile(
169 result = compiled_model(*example_args)
178 model.parameters(), compiled_model.parameters()
714 compiled_model = torch.compile(
725 result = compiled_model(*example_args)
730 result = compiled_model(*example_args)
747 compiled_model = torch.compile(
753 result = compiled_model(data, ref_data)
770 compiled_model = torch.compile(
781 result = compiled_model()
[all …]
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor_package.py35 compiled_model = load_package(package_path, device)
36 return compiled_model
62 compiled_model = compile(
70 actual = compiled_model(*example_inputs)
H A Dtest_control_flow.py184 compiled_model = torch.compile(backend=cnt, fullgraph=True)(model)
204 result_compiled = compiled_model(*inputs_with_predicates)
594 compiled_model = torch.compile(backend=cnt, fullgraph=True)(model)
616 result_compiled = compiled_model(*inputs_with_counters)
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
H A Dtest_util.cc104 std::unique_ptr<CompiledModel> compiled_model; in Invoke() local
107 &compiled_model)); in Invoke()
112 RETURN_IF_ERROR(compiled_model->NewRun( in Invoke()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/
H A Dapi.cc384 std::unique_ptr<CompiledModel>* compiled_model) { in Compile() argument
399 *compiled_model = std::move(compiled_model_impl); in Compile()
406 std::unique_ptr<CompiledModel>* compiled_model) { in ReadSerializedModel() argument
416 *compiled_model = std::move(compiled_model_impl); in ReadSerializedModel()
H A Dapi.h73 std::unique_ptr<CompiledModel>* compiled_model);
80 std::unique_ptr<CompiledModel>* compiled_model);
H A DBUILD100 srcs = ["compiled_model.fbs"],
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dgl_delegate.cc266 std::unique_ptr<CompiledModel> compiled_model; in Prepare() local
268 *workgroups_calculator, &compiled_model)); in Prepare()
272 RETURN_IF_ERROR(compiled_model->NewRun(runtime_options, &phwc4_objects_, in Prepare()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_autograd_function.py405 compiled_model = torch._dynamo.optimize("eager")(mod)
406 after = compiled_model(*args, **kwargs)
415 compiled_model = torch._dynamo.optimize("eager")(mod)
416 after = compiled_model(*args, **kwargs)
694 compiled_model = torch._dynamo.optimize("eager")(mod)
695 after = compiled_model(*args, **kwargs)
H A Dtest_model_output.py350 compiled_model = torch.compile(model, backend="eager")
351 compiled_result = compiled_model(sequence_output)
H A Dtest_modules.py2345 compiled_model = torch.compile(model, backend="aot_eager")
2353 pred = compiled_model(x)
2912 compiled_model = torch.compile(model2, backend="eager")
2913 compiled_model.x = torch.tensor([[100, 100, 100, 100], [200, 200, 200, 200]])
2914 compiled_model(input)
2916 self.assertEqual(model.x, compiled_model.x)
H A Dtest_misc.py7451 compiled_model = torch.compile(model)
7453 model.forward = compiled_model.dynamo_ctx(model.forward)
7454 model.training_step = compiled_model.dynamo_ctx(model.training_step)
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_dtensor_compile.py709 compiled_model = torch.compile(model, backend=cnt, fullgraph=True)
711 out = compiled_model(inp)
715 code = run_and_get_triton_code(compiled_model, inp)