Searched refs:ETKernelIndex (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/executorch/ |
| H A D | model.py | 140 class ETKernelIndex: class
176 ) -> ETKernelIndex:
180 ETKernelIndex.grow_from_backend_indices(kernel_index, backend_indices)
181 return ETKernelIndex(kernel_index)
185 ) -> ETKernelIndex:
186 ETKernelIndex.grow_from_backend_indices(self.index, backend_indices)
213 def merge_indices(index_a: ETKernelIndex, index_b: ETKernelIndex) -> ETKernelIndex: argument
220 return ETKernelIndex(combined)
|
| H A D | parse.py | 8 from torchgen.executorch.model import ETKernelIndex, ETKernelKey
76 def parse_et_yaml_struct(es: object) -> ETKernelIndex:
97 return ETKernelIndex(indices)
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | gen_executorch.py | 28 from torchgen.executorch.model import ETKernelIndex, ETKernelKey, ETParsedYaml
293 kernel_index: ETKernelIndex, argument
333 g: NativeFunctionsGroup | NativeFunction, kernel_index: ETKernelIndex argument
356 kernel_index: ETKernelIndex, argument
410 kernel_index: ETKernelIndex, argument
414 ETKernelIndex,
444 kernel_index: ETKernelIndex, argument
542 kernel_index: ETKernelIndex, argument
705 dict[DispatchKey, dict[OperatorName, BackendMetadata]] | ETKernelIndex,
738 return native_functions, ETKernelIndex(index=filtered_index)
[all …]
|
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_executorch_custom_ops.py | 12 from torchgen.executorch.model import ETKernelIndex
123 kernel_index=ETKernelIndex(index={}),
143 kernel_index=ETKernelIndex(index={}),
|
| H A D | test_executorch_gen.py | 9 from torchgen.executorch.model import ETKernelIndex, ETKernelKey
365 self.kernel_index = ETKernelIndex.from_backend_indices(backend_indices)
|
| /aosp_15_r20/external/pytorch/torchgen/executorch/api/ |
| H A D | custom_ops.py | 18 from torchgen.executorch.model import ETKernelIndex
87 kernel_index: ETKernelIndex, argument
|