Searched defs:OpsetVersion (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/onnx/_internal/ |
| H A D | registration.py | 20 OpsetVersion = int variable
24 target: OpsetVersion, registered_opsets: Collection[OpsetVersion]
144 def __getitem__(self, key: OpsetVersion) -> Callable:
152 def get(self, opset: OpsetVersion) -> Optional[Callable]:
160 def add(self, func: Callable, opset: OpsetVersion) -> None:
176 def add_custom(self, func: Callable, opset: OpsetVersion) -> None:
185 def remove_custom(self, opset: OpsetVersion) -> None:
215 self, name: str, opset: OpsetVersion, func: Callable, custom: bool = False
241 def unregister(self, name: str, opset: OpsetVersion) -> None:
270 opset: Union[OpsetVersion, Sequence[OpsetVersion]],
[all …]
|