| /aosp_15_r20/external/executorch/backends/apple/coreml/compiler/ |
| H A D | coreml_preprocess.py | 89 def model_type_from_compile_specs(compile_specs: List[CompileSpec]) -> MODEL_TYPE:
93 for compile_spec in compile_specs:
114 compile_specs: List[CompileSpec],
119 for compile_spec in compile_specs:
140 compile_specs: List[CompileSpec],
145 for compile_spec in compile_specs:
154 compile_specs: List[CompileSpec],
159 for compile_spec in compile_specs:
195 compile_specs: List[CompileSpec],
200 for compile_spec in compile_specs:
[all …]
|
| /aosp_15_r20/external/executorch/examples/apple/coreml/scripts/ |
| H A D | export.py | 89 def lower_module_to_coreml(module, compile_specs, example_inputs): argument
101 compile_specs,
168 compile_specs = generate_compile_specs_from_args(args)
178 skip_ops_for_coreml_delegation=None, compile_specs=compile_specs
188 compile_specs=compile_specs,
|
| H A D | inspector_utils.py | 126 compile_specs: List[CompileSpec],
132 compile_specs=compile_specs,
268 compile_specs: List[CompileSpec],
279 compile_specs=compile_specs,
|
| /aosp_15_r20/external/executorch/extension/llm/export/ |
| H A D | partitioner_lib.py | 63 compile_specs = [CompileSpec("use_fp16", bytes([True]))]
64 return MPSPartitioner(compile_specs) # pyre-fixme[16]
131 compile_specs = CoreMLBackend.generate_compile_specs( # pyre-fixme[16]
143 compile_specs=compile_specs,
|
| /aosp_15_r20/external/executorch/backends/vulkan/partitioner/ |
| H A D | vulkan_partitioner.py | 271 compile_specs = []
276 compile_specs.append(CompileSpec(key, value_bytes))
280 compile_specs.append(CompileSpec(key, value_bytes))
283 compile_specs.append(
288 compile_specs.append(
293 compile_specs.append(
301 return compile_specs
|
| /aosp_15_r20/external/executorch/backends/apple/mps/partition/ |
| H A D | mps_partitioner.py | 55 def __init__(self, compile_specs: List[CompileSpec]) -> None:
56 self.compile_specs = compile_specs
57 self.delegation_spec = DelegationSpec(MPSBackend.__name__, compile_specs)
62 edge_program=edge_program, compiler_specs=self.delegation_spec.compile_specs
|
| /aosp_15_r20/external/executorch/backends/arm/test/misc/ |
| H A D | test_tosa_spec.py | 95 def test_create_from_compilespec(self, compile_specs: list[CompileSpec]):
96 tosa_spec = TosaSpecification.create_from_compilespecs(compile_specs)
100 def test_create_from_invalid_compilespec(self, compile_specs: list[CompileSpec]):
103 tosa_spec = TosaSpecification.create_from_compilespecs(compile_specs)
|
| /aosp_15_r20/external/executorch/backends/apple/coreml/partition/ |
| H A D | coreml_partitioner.py | 63 compile_specs: Optional[List[CompileSpec]] = None,
71 compile_specs=compile_specs if compile_specs is not None else [],
|
| /aosp_15_r20/external/executorch/exir/backend/ |
| H A D | backend_api.py | 70 compile_specs: List[CompileSpec],
115 compile_specs,
121 compile_specs=compile_specs,
252 delegation_spec.compile_specs,
|
| /aosp_15_r20/external/executorch/runtime/executor/test/ |
| H A D | backend_integration_test.cpp | 85 ArrayRef<CompileSpec> compile_specs) const override { in init()
87 return init_fn_.value()(processed, compile_specs, context); in init()
353 ET_UNUSED ArrayRef<CompileSpec> compile_specs, in TEST_P()
397 ET_UNUSED ArrayRef<CompileSpec> compile_specs, in TEST_P()
496 ET_UNUSED ArrayRef<CompileSpec> compile_specs, in TEST_P()
539 ET_UNUSED ArrayRef<CompileSpec> compile_specs, in TEST_P()
652 ET_UNUSED ArrayRef<CompileSpec> compile_specs, in TEST_P()
|
| H A D | test_backend_with_delegate_mapping.cpp | 94 ArrayRef<CompileSpec> compile_specs) const override { in init()
96 (void)compile_specs; in init()
|
| /aosp_15_r20/external/executorch/examples/apple/mps/scripts/ |
| H A D | mps_example.py | 178 compile_specs = [CompileSpec("use_fp16", bytes([args.use_fp16]))] variable
182 edge = edge.to_backend(MPSPartitioner(compile_specs=compile_specs))
190 MPSBackend.__name__, edge.exported_program(), compile_specs
|
| /aosp_15_r20/external/executorch/backends/apple/coreml/runtime/test/ |
| H A D | export_stateful_model.py | 52 compile_specs = CoreMLBackend.generate_compile_specs(
60 compile_specs=compile_specs,
|
| /aosp_15_r20/external/executorch/exir/backend/test/ |
| H A D | test_backends.py | 135 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
137 "BackendWithCompilerDemo", edgeir_m.exported_program, compile_specs
235 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
237 "BackendWithCompilerDemo", edge_graph_module.exported_program, compile_specs
289 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
291 "BackendWithCompilerDemo", edgeir_m.exported_program, compile_specs
358 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
360 "BackendWithCompilerDemo", edgeir_m.exported_program, compile_specs
505 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
509 compile_specs,
|
| H A D | test_backends_lifted.py | 159 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
161 "BackendWithCompilerDemo", edgeir_m.exported_program(), compile_specs
253 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
257 compile_specs,
303 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
305 "BackendWithCompilerDemo", edgeir_m.exported_program(), compile_specs
366 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
368 "BackendWithCompilerDemo", edgeir_m.exported_program(), compile_specs
505 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
509 compile_specs,
|
| H A D | test_compatibility.py | 37 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
39 BackendWithCompilerDemo.__name__, edgeir_m.exported_program(), compile_specs
|
| /aosp_15_r20/external/executorch/backends/apple/mps/test/ |
| H A D | test_mps_utils.py | 229 compile_specs = [CompileSpec("use_fp16", bytes([use_fp16]))]
235 MPSPartitioner(compile_specs=compile_specs)
246 MPSBackend.__name__, edge_program.exported_program(), compile_specs
|
| /aosp_15_r20/external/executorch/backends/apple/coreml/test/ |
| H A D | test_coreml_partitioner.py | 119 compile_specs = CoreMLBackend.generate_compile_specs(
122 partitioner = CoreMLPartitioner(compile_specs=compile_specs)
|
| /aosp_15_r20/external/executorch/exir/backend/test/demos/ |
| H A D | test_delegate_aten_mode.py | 40 compile_specs = [CompileSpec("max_value", bytes([max_value]))]
44 compile_specs,
|
| /aosp_15_r20/external/executorch/backends/arm/ |
| H A D | arm_partitioner.py | 50 self.delegation_spec.compile_specs
55 for spec in self.delegation_spec.compile_specs:
|
| /aosp_15_r20/external/executorch/backends/qualcomm/ |
| H A D | qnn_preprocess.py | 42 compile_specs: List[CompileSpec],
44 option = generate_qnn_executorch_option(compile_specs)
|
| /aosp_15_r20/external/executorch/backends/qualcomm/utils/ |
| H A D | utils.py | 225 options = flatbuffer_to_option(m.compile_specs[0].value)
232 m.compile_specs[0].value, m.processed_bytes
244 module.compile_specs[0].value, module.processed_bytes
250 module: flatbuffer_to_option(module.compile_specs[0].value)
263 module.compile_specs[0] = spec
|
| /aosp_15_r20/external/executorch/exir/serde/ |
| H A D | serialize.py | 262 for cs in lowered_module.compile_specs
277 compile_specs=serialized_compile_spec,
547 compile_specs = [
549 for cs in serialized_lowered_module.compile_specs
563 compile_specs,
|
| /aosp_15_r20/external/executorch/backends/apple/mps/runtime/ |
| H A D | MPSCompiler.mm | 45 ArrayRef<CompileSpec> compile_specs) {
46 MPS_UNUSED(compile_specs);
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/ |
| H A D | VulkanBackend.cpp | 121 GraphConfig get_graph_config(ArrayRef<CompileSpec>& compile_specs) { in get_graph_config() argument
124 for (const CompileSpec& spec : compile_specs) { in get_graph_config()
522 ArrayRef<CompileSpec> compile_specs) const override { in init()
526 new (compute_graph) ComputeGraph(get_graph_config(compile_specs)); in init()
|