Searched defs:SoftmaxWorkerTask (Results 1 – 1 of 1) sorted by relevance
3681 struct SoftmaxWorkerTask : cpu_backend_threadpool::Task { struct3682 SoftmaxWorkerTask(const SoftmaxParams& params, in SoftmaxWorkerTask() argument3693 void Run() override { in Run()3700 const RuntimeShape& input_shape;3701 const float* input_data;3702 const RuntimeShape& output_shape;3703 float* output_data;3704 int start_batch;3705 int end_batch;