| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/stable_diffusion/ |
| H A D | qaihub_stable_diffusion.py | 61 compiler_specs, argument
65 compiler_specs=compiler_specs,
73 compiler_specs=compiler_specs,
81 compiler_specs=compiler_specs,
215 def inference(args, compiler_specs, pte_files): argument
239 compiler_specs,
362 compiler_specs = generate_qnn_executorch_compiler_spec(
401 inference(args, compiler_specs, pte_files)
|
| /aosp_15_r20/external/executorch/backends/qualcomm/tests/ |
| H A D | test_qnn_delegate.py | 68 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
529 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
681 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
1272 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
1480 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
1489 TestQNN.compiler_specs = generate_qnn_executorch_compiler_spec(
1534 compiler_specs = generate_qnn_executorch_compiler_spec(
1538 partitioner = QnnPartitioner(compiler_specs)
1550 compiler_specs = generate_qnn_executorch_compiler_spec(
1555 compiler_specs=compiler_specs,
[all …]
|
| H A D | utils.py | 118 compiler_specs: List[CompileSpec] = None variable in TestQNN
327 self.compiler_specs, skip_node_id_set, skip_node_op_set
|
| H A D | models.py | 144 compiler_specs, argument
165 partitioner = partitioner_type(compiler_specs)
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/utils/ |
| H A D | utils.py | 17 def preprocess_binary(ctx_bin, compiler_specs): argument
19 generate_qnn_executorch_option(compiler_specs),
26 compiler_specs: str,
34 ctx_bin = preprocess_binary(f.read(), compiler_specs)
36 generate_qnn_executorch_option(compiler_specs), ctx_bin
|
| H A D | export.py | 47 def get_io_info(prog_info, ctx_bin_path, compiler_specs): argument
72 ctx_bin = preprocess_binary(f.read(), compiler_specs)
75 generate_qnn_executorch_option(compiler_specs), ctx_bin
216 compiler_specs = generate_qnn_executorch_compiler_spec(
275 graph_info = get_io_info(prog_info, ctx_bin, compiler_specs)
|
| /aosp_15_r20/external/executorch/backends/qualcomm/utils/ |
| H A D | utils.py | 686 def preprocess_binary(ctx_bin, compiler_specs): argument
688 generate_qnn_executorch_option(compiler_specs),
694 compiler_specs = generate_qnn_executorch_compiler_spec(
703 else preprocess_binary(Path(f"{ctx_path}").read_bytes(), compiler_specs)
722 generate_qnn_executorch_option(compiler_specs),
748 QnnPartitioner(compiler_specs)
760 compiler_specs: List[CompileSpec],
767 generate_qnn_executorch_option(compiler_specs), processed_bytes
779 compiler_options = flatbuffer_to_option(compiler_specs[0].value)
809 return edge_prog_mgr.to_backend(QnnPartitioner(compiler_specs)).to_executorch(
|
| /aosp_15_r20/external/executorch/backends/qualcomm/partition/ |
| H A D | qnn_partitioner.py | 42 compiler_specs, argument
52 generate_qnn_executorch_option(compiler_specs)
108 compiler_specs: List[CompileSpec],
112 self.compiler_specs_snapshot = copy.deepcopy(compiler_specs)
|
| H A D | utils.py | 15 compiler_specs: List[CompileSpec],
17 for compiler_spec in compiler_specs:
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/llama2/ |
| H A D | qaihub_llama2_7b.py | 51 compiler_specs = generate_qnn_executorch_compiler_spec(
110 compiler_specs=compiler_specs,
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/llama3/ |
| H A D | qaihub_llama3_8b.py | 52 compiler_specs = generate_qnn_executorch_compiler_spec(
112 compiler_specs=compiler_specs,
|
| /aosp_15_r20/external/executorch/backends/apple/mps/partition/ |
| H A D | mps_partitioner.py | 39 def __init__(self, edge_program: torch.export.ExportedProgram, compiler_specs): argument
62 edge_program=edge_program, compiler_specs=self.delegation_spec.compile_specs
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/ |
| H A D | llama.py | 198 compiler_specs = generate_qnn_executorch_compiler_spec(
205 compiler_specs, skip_node_op_set=skip_node_op_set
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama2/ |
| H A D | llama.py | 313 compiler_specs = generate_qnn_executorch_compiler_spec(
318 partitioner = QnnPartitioner(compiler_specs)
|