template class habana::v1::HpuKernel

Overview

Final class for all HpuKernels. More…

#include <hpu_kernel.h>

template <class Functor>
class HpuKernel: public habana::v1::HpuBaseKernel {
public:
    // methods

    virtual BaseFunctor& GetFunctor();
};

Inherited Members

public:
    // methods

    virtual void Compute(tensorflow::OpKernelContext* context);
    virtual BaseFunctor& GetFunctor() = 0;
    virtual HpuKernelImpl* Pimpl();

Detailed Documentation

Final class for all HpuKernels.

In order to register new kernel, BaseFunctor needs to be implemented, i.e. HpuKernel<SomeFunctor>.

Methods

virtual BaseFunctor& GetFunctor()

Getter for Functor - core part of any HpuKernel.