Searched refs:ETKernelKey (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torchgen/executorch/ |
| D | model.py | 54 class ETKernelKey: class
68 ) -> list[ETKernelKey]:
124 kernel_keys.append(ETKernelKey(tuple(arg_list)))
141 index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]]
149 ) -> dict[ETKernelKey, BackendMetadata]:
162 kernel_index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]], argument
169 kernel_index[op][ETKernelKey(default=True)] = backend_metadata
171 kernel_index[op] = {ETKernelKey(default=True): backend_metadata}
178 OperatorName, dict[ETKernelKey, BackendMetadata]
200 ETKernelKey(default=True),
|
| D | parse.py | 8 from torchgen.executorch.model import ETKernelIndex, ETKernelKey
27 def parse_from_yaml(ei: dict[str, object]) -> dict[ETKernelKey, BackendMetadata]:
43 kernel_mapping: dict[ETKernelKey, BackendMetadata] = {}
62 [ETKernelKey((), default=True)]
64 … else ETKernelKey.gen_from_yaml(arg_meta, type_alias, dim_order_alias) # type: ignore[arg-type]
81 indices: dict[OperatorName, dict[ETKernelKey, BackendMetadata]] = {}
|
| /external/pytorch/tools/test/ |
| D | test_executorch_gen.py | 9 from torchgen.executorch.model import ETKernelIndex, ETKernelKey
466 self.default_kernel_key = ETKernelKey(default=True)
476 specialized_kernel_key = ETKernelKey.gen_from_yaml(
520 specialized_kernel_key = ETKernelKey.gen_from_yaml(
544 specialized_kernel_key = ETKernelKey.gen_from_yaml(
|
| /external/pytorch/torchgen/ |
| D | gen_executorch.py | 28 from torchgen.executorch.model import ETKernelIndex, ETKernelKey, ETParsedYaml
187 unbox_kernel_entry: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]], argument
190 kernel_key: ETKernelKey | list[ETKernelKey] = unbox_kernel_entry[1][0]
298 item: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]] argument
302 items: list[tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]]] = [
|