U yh@sFUddlmZdddgZGdddZeZeed<GdddZdS))AbstractImplHolderSimpleLibraryRegistrySimpleOperatorEntry singletonc@s(eZdZdZddZeddddZdS) raJRegistry for the "simple" torch.library APIs The "simple" torch.library APIs are a higher-level API on top of the raw PyTorch DispatchKey registration APIs that includes: - fake impl Registrations for these APIs do not go into the PyTorch dispatcher's table because they may not directly involve a DispatchKey. For example, the fake impl is a Python function that gets invoked by FakeTensor. Instead, we manage them here. SimpleLibraryRegistry is a mapping from a fully qualified operator name (including the overload) to SimpleOperatorEntry. cCs i|_dSN)_data)selfr P/var/www/html/venv/lib/python3.8/site-packages/torch/_library/simple_registry.py__init__szSimpleLibraryRegistry.__init__r)qualnamereturncCs"||jkrt||j|<|j|Sr)rrrr r r r finds zSimpleLibraryRegistry.findN)__name__ __module__ __qualname____doc__r strrr r r r rsc@seZdZdZedddZdS)rzThis is 1:1 to an operator overload. The fields of SimpleOperatorEntry are Holders where kernels can be registered to. )r cCs||_t||_dSr)r r abstract_implrr r r r *szSimpleOperatorEntry.__init__N)rrrrrr r r r r r#sN)rr__all__rr__annotations__rr r r r s