Searched refs:OperatorEntry (Results 1 – 8 of 8) sorted by relevance
21 OperatorEntry::OperatorEntry(OperatorName&& operator_name) in OperatorEntry() function in c10::impl::OperatorEntry58 const AnnotatedKernel& OperatorEntry::missingKernel() const { in missingKernel()63 const AnnotatedKernel& OperatorEntry::ambiguousAutogradOtherKernel() const { in ambiguousAutogradOtherKernel()69 void OperatorEntry::assertSignatureIsCorrect(const CppSignature& call_signature, bool has_symint) c… in assertSignatureIsCorrect()81 void OperatorEntry::registerSchema(FunctionSchema&& schema, std::string&& debug, std::vector<at::Ta… in registerSchema()98 void OperatorEntry::deregisterSchema() { in deregisterSchema()104 OperatorEntry::AnnotatedKernelContainerIterator OperatorEntry::registerKernel( in registerKernel()184 void OperatorEntry::deregisterKernel_( in registerKernel()206 void OperatorEntry::updateFallback(const c10::Dispatcher& dispatcher, DispatchKey dispatch_key) { in registerKernel()210 const KernelFunction& OperatorEntry::computeDispatchTableEntry(const c10::Dispatcher& dispatcher, D… in registerKernel()[all …]
70 class TORCH_API OperatorEntry final {72 explicit OperatorEntry(OperatorName&& operator_name);74 OperatorEntry(const OperatorEntry&) = delete;75 OperatorEntry(OperatorEntry&&) noexcept = delete;76 OperatorEntry& operator=(const OperatorEntry&) = delete;77 OperatorEntry& operator=(OperatorEntry&&) noexcept = delete;
69 friend class impl::OperatorEntry;75 impl::OperatorEntry op;325 impl::OperatorEntry::AnnotatedKernelContainerIterator kernel_handle);
373 …OperatorName& op_name, std::optional<DispatchKey> dispatch_key, impl::OperatorEntry::AnnotatedKern… in deregisterImpl_()
82 …"[W OperatorEntry.cpp:133] Warning: Overriding a previously registered kernel for the same operato…
699 # OperatorEntry::dispatchTable_ to reduce peak memory during library
2126 "aten/src/ATen/core/dispatch/OperatorEntry.cpp",
1056 "aten/src/ATen/core/dispatch/OperatorEntry.cpp",