Searched refs:StackFp16CPUKernel (Results 1 – 2 of 2) sorted by relevance
37 void StackFp16CPUKernel::InitMallocFlags() { in InitMallocFlags()45 int StackFp16CPUKernel::MallocAssignBuffer() { in MallocAssignBuffer()63 void StackFp16CPUKernel::FreeBuffer() { in FreeBuffer()76 int StackFp16CPUKernel::Init() { in Init()86 int StackFp16CPUKernel::DoExecute(int task_id) { in DoExecute()101 auto stack = reinterpret_cast<StackFp16CPUKernel *>(cdata); in StackRun()108 int StackFp16CPUKernel::Run() { in Run()133 REG_KERNEL(kCPU, kNumberTypeFloat16, PrimitiveType_Stack, LiteKernelCreator<StackFp16CPUKernel>)
24 class StackFp16CPUKernel : public StackBaseCPUKernel {26 StackFp16CPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in StackFp16CPUKernel() function29 ~StackFp16CPUKernel() override = default;