Class which, on construction, registers an operator in the dispatch table. More...

Public Types
using	OpHandleGetter) = const OperatorHandle &(

Public Member Functions
	KernelRegistrar (OpHandleGetter op, TensorTypeId dispatch_key, KernelFunction kernel, KernelCacheCreatorFunction *cache_creator)

	KernelRegistrar (KernelRegistrar &&rhs)

KernelRegistrar &	operator= (KernelRegistrar &&rhs)=delete

Detailed Description

Class which, on construction, registers an operator in the dispatch table.

The intent is that this class is constructed at static initialization time so that operators automatically get registered when a dlopen() occurs.

You shouldn't call this directly; instead, use the C10_REGISTER_KERNEL macros.

Definition at line 33 of file KernelRegistration.h.

Constructor & Destructor Documentation

c10::KernelRegistrar::KernelRegistrar	(	OpHandleGetter *	op,
		TensorTypeId	dispatch_key,
		KernelFunction *	kernel,
		KernelCacheCreatorFunction *	cache_creator
	)

inlineexplicit

Parameters

op	The operator to register the kernel for
dispatch_key	The dispatch key to register the function to
kernel	The concrete function implementation to register
cache_creator	A function initializing the cache for the kernel

Definition at line 43 of file KernelRegistration.h.

The documentation for this class was generated from the following file: