Searched full:et_kernel_metadata (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/executorch/codegen/tools/ |
| H A D | gen_oplist.py | 42 # et_kernel_metadata:
158 et_kernel_metadata: Optional[Dict[str, List[str]]] = None,
182 output["et_kernel_metadata"] = et_kernel_metadata
211 et_kernel_metadata = {}
219 et_kernel_metadata = merge_et_kernel_metadata(
220 et_kernel_metadata, {op: ["default"] for op in op_set}
227 et_kernel_metadata = merge_et_kernel_metadata(
228 et_kernel_metadata, {op: op_metadata}
236 et_kernel_metadata = merge_et_kernel_metadata(
237 et_kernel_metadata, _get_kernel_metadata_for_model(model_file_path)
[all …]
|
| H A D | gen_selected_op_variants.py | 67 # Collect et_kernel_metadata from selected_operators.yaml and extract dtypes
70 et_kernel_metadata = selected_operators_dict.get("et_kernel_metadata", {})
71 assert isinstance(et_kernel_metadata, dict)
74 for operator_name, kernel_metadata_str in et_kernel_metadata.items():
|
| /aosp_15_r20/external/pytorch/torchgen/selective_build/ |
| H A D | selector.py | 52 et_kernel_metadata: dict[str, list[str]]
82 "et_kernel_metadata",
117 et_kernel_metadata = data.get("et_kernel_metadata", {})
118 assert isinstance(et_kernel_metadata, dict)
136 et_kernel_metadata,
244 if op_name not in self.et_kernel_metadata:
250 for model_kernel_keys in self.et_kernel_metadata[op_name]:
286 ret["et_kernel_metadata"] = self.et_kernel_metadata
329 et_kernel_metadata = merge_et_kernel_metadata(
330 lhs.et_kernel_metadata, rhs.et_kernel_metadata
[all …]
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_gen.py | 484 "et_kernel_metadata": {
528 "et_kernel_metadata": {
551 "et_kernel_metadata": {
575 "et_kernel_metadata": {
613 "et_kernel_metadata": {"custom_1::op_1": ["default"]},
|
| H A D | test_selective_build.py | 306 et_kernel_metadata:
|
| /aosp_15_r20/external/executorch/codegen/tools/test/ |
| H A D | test_gen_selected_op_variants.py | 40 et_kernel_metadata:
99 et_kernel_metadata: {}
|
| /aosp_15_r20/external/pytorch/test/edge/ |
| H A D | selected_operators.yaml | 6 et_kernel_metadata:
|