Searched refs:SoftmaxInt8CPUKernel (Results 1 – 2 of 2) sorted by relevance
32 SoftmaxInt8CPUKernel::~SoftmaxInt8CPUKernel() { in ~SoftmaxInt8CPUKernel()39 int SoftmaxInt8CPUKernel::Init() { in Init()80 int SoftmaxInt8CPUKernel::ReSize() { return SoftmaxBaseCPUKernel::ReSize(); } in ReSize()82 int SoftmaxInt8CPUKernel::DoSoftmax(int task_id) { in DoSoftmax()118 auto softmax_kernel = reinterpret_cast<SoftmaxInt8CPUKernel *>(cdata); in SoftmaxRun()127 int SoftmaxInt8CPUKernel::Run() { in Run()154 REG_KERNEL(kCPU, kNumberTypeInt8, PrimitiveType_Softmax, LiteKernelCreator<SoftmaxInt8CPUKernel>)
25 class SoftmaxInt8CPUKernel : public SoftmaxBaseCPUKernel {27 SoftmaxInt8CPUKernel(OpParameter *parameter, const std::vector<lite::Tensor *> &inputs, in SoftmaxInt8CPUKernel() function30 ~SoftmaxInt8CPUKernel() override;