Searched defs:SchemaKind (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torchgen/dest/ |
| H A D | register_dispatch_key.py | 584 self, k: SchemaKind, parent_class: str, generate_super: bool
612 def gen_class_set_output_body(self, k: SchemaKind, maybe_create_proxy: bool) -> str:
671 def gen_class_ctor(self, k: SchemaKind, class_name: str, returns: int) -> str:
691 k: SchemaKind,
|
| /aosp_15_r20/external/executorch/exir/dialects/edge/op/ |
| H A D | api.py | 36 def to_variant(op: OpOverload, variant: SchemaKind) -> OpOverload:
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | native_function_generation.py | 270 f: NativeFunction, k: SchemaKind
|
| H A D | gen.py | 1365 d: dict[ViewSchemaKind | SchemaKind, NativeFunction]
1412 d: dict[SchemaKind, NativeFunction]
|
| H A D | model.py | 1085 class SchemaKind(Enum): class
1192 def from_dict(d: dict[SchemaKind, NativeFunction]) -> NativeFunctionsGroup | None:
|