Searched defs:BackendMetadata (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/torchgen/executorch/ |
H A D | model.py | 162 kernel_index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]], argument 175 backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]] argument 184 self, backend_indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]] argument
|
/aosp_15_r20/external/pytorch/torchgen/ |
H A D | gen_executorch.py | 187 unbox_kernel_entry: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]], argument 298 item: tuple[NativeFunction, tuple[ETKernelKey, BackendMetadata]] argument 343 def gen_decl(metadata: BackendMetadata, include_context: bool) -> str: 742 m: dict[OperatorName, BackendMetadata] argument
|
H A D | native_function_generation.py | 381 indices: dict[DispatchKey, dict[OperatorName, BackendMetadata]], argument
|
H A D | model.py | 1217 class BackendMetadata: class 1284 parent_index: dict[DispatchKey, dict[OperatorName, BackendMetadata]], argument
|
/aosp_15_r20/external/pytorch/torchgen/dest/ |
H A D | lazy_ir.py | 464 metadata: BackendMetadata,
|
/aosp_15_r20/external/pytorch/test/cpp/api/ |
H A D | tensor.cpp | 1221 TEST(TensorTest, BackendMetadata) { in TEST() argument
|