Searched defs:SoftmaxWorkerTask (Results 1 – 1 of 1) sorted by relevance
3672 struct SoftmaxWorkerTask : cpu_backend_threadpool::Task { struct3673 SoftmaxWorkerTask(const SoftmaxParams& params, in SoftmaxWorkerTask() argument3684 void Run() override { in Run()3691 const RuntimeShape& input_shape;3692 const float* input_data;3693 const RuntimeShape& output_shape;3694 float* output_data;3695 int start_batch;3696 int end_batch;