Caffe2 - C++ API
A deep learning, cross platform ML framework
Data Fields
c10::DispatchTableEntry Struct Referencefinal

The dispatch table stores a pointer to a kernel function and a pointer to a function initializing a cache for the kernel. More...

#include <DispatchTable.h>

Data Fields

KernelFunctionkernel_func
 
KernelCacheCreatorFunctioncache_creator_func
 

Detailed Description

The dispatch table stores a pointer to a kernel function and a pointer to a function initializing a cache for the kernel.

If the kernel wants to use the cache, they supply the state initializer when the kernel is registered. When a kernel is looked up from the dispatcher, a new cache instance is created for it and each call to that kernel will get this same cache instance.

Definition at line 35 of file DispatchTable.h.


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