class tensorflow::register_kernel::v1::ForHpuWithName

Overview

KernelDefBuilder class that should be used to register HpuKernels. More…

#include <hpu_kernel.h>

class ForHpuWithName: public KernelDefBuilder {
public:
    // construction

    ForHpuWithName(const char* op);
};

Detailed Documentation

KernelDefBuilder class that should be used to register HpuKernels.

It resides in tensorflow::register_kernel namespace to be compatible with macro REGISTER_KERNEL_BUILDER from tensorflow/core/framework/op_kernel.h.

Example: REGISTER_KERNEL_BUILDER(ForHpuWithName (“CustomOp”).TypeConstraint<float>(“T”), habana::HpuKernel<CustomFunctor<float>>);

It provides following functionalities:

  • KernelDefBuilder Device is set to HPU

  • automatic clustering

Construction

ForHpuWithName(const char* op)

C’tor.