Searched refs:DepthToSpaceFp16CPUKernel (Results 1 – 2 of 2) sorted by relevance
23 class DepthToSpaceFp16CPUKernel : public DepthToSpaceCPUKernel {25 DepthToSpaceFp16CPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in DepthToSpaceFp16CPUKernel() function28 ~DepthToSpaceFp16CPUKernel() = default;
25 int DepthToSpaceFp16CPUKernel::Init() { in Init()35 …kCPU, kNumberTypeFloat16, PrimitiveType_DepthToSpace, LiteKernelCreator<DepthToSpaceFp16CPUKernel>)