Home
last modified time | relevance | path

Searched refs:export_model (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/executorch/exir/verification/test/
H A Dtest_verifier.py85 export_model = export(m, example_input)
92 export_model, compile_config=compile_config_without_edge_op
133 export_model = export(m, example_input)
139 export_model, compile_config=compile_config_with_dim_order
142 export_model, compile_config=compile_config_with_stride
/aosp_15_r20/external/executorch/extension/training/examples/XOR/
H A Dtargets.bzl36 srcs = ["export_model.py"],
46 name = "export_model",
47 main_module = "executorch.extension.training.examples.XOR.export_model",
/aosp_15_r20/external/executorch/examples/cadence/models/
H A Dvision_transformer.py16 from executorch.backends.cadence.aot.export_example import export_model
28 export_model(model, example_inputs)
H A Dmobilenet_v2.py16 from executorch.backends.cadence.aot.export_example import export_model
30 export_model(model, example_inputs)
H A Dresnet50.py16 from executorch.backends.cadence.aot.export_example import export_model
30 export_model(model, example_inputs)
H A Dbabyllama.py15 from executorch.backends.cadence.aot.export_example import export_model
38 export_model(model, example_inputs)
H A Dwav2vec2.py15 from executorch.backends.cadence.aot.export_example import export_model
61 export_model(model, example_inputs)
H A Drnnt_joiner.py17 from executorch.backends.cadence.aot.export_example import export_model
65 export_model(model, example_inputs)
H A Drnnt_predictor.py17 from executorch.backends.cadence.aot.export_example import export_model
69 export_model(model, example_inputs)
H A Drnnt_encoder.py17 from executorch.backends.cadence.aot.export_example import export_model
161 export_model(encoder, example_inputs)
/aosp_15_r20/external/executorch/build/
H A Dtest_android_ci.sh11 export_model() { function
29 export_model
/aosp_15_r20/external/executorch/examples/cadence/operators/
H A Dquantized_linear_op.py15 from executorch.backends.cadence.aot.export_example import export_model
42 export_model(model, example_inputs)
H A Dquantized_conv1d_op.py15 from executorch.backends.cadence.aot.export_example import export_model
58 export_model(model, example_inputs)
/aosp_15_r20/external/executorch/examples/models/phi-3-mini-lora/
H A DREADME.md16 python export_model.py
/aosp_15_r20/external/executorch/backends/cadence/aot/
H A Dexport_example.py59 def export_model( function
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_mkldnn_pattern_matcher.py137 export_model = capture_pre_autograd_graph(
145 prepare_qat_pt2e(export_model, quantizer)
147 else prepare_pt2e(export_model, quantizer)
/aosp_15_r20/external/executorch/extension/training/
H A DREADME.md44 python3 extension/training/examples/XOR/export_model.py --outdir /tmp/foobar
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py2956 export_model = capture_pre_autograd_graph(
2964 prepare_qat_pt2e(export_model, quantizer)
2966 else prepare_pt2e(export_model, quantizer)
/aosp_15_r20/external/pytorch/test/quantization/pt2e/
H A Dtest_x86inductor_quantizer.py559 export_model = m if is_qat else copy.deepcopy(m)
578 return export_model, prepare_model, convert_model