Searched defs:SoftmaxWorkerTask (Results 1 – 1 of 1) sorted by relevance
3356 struct SoftmaxWorkerTask : cpu_backend_threadpool::Task { struct3357 SoftmaxWorkerTask(const SoftmaxParams& params, in SoftmaxWorkerTask() argument3368 void Run() override { in Run()3375 const RuntimeShape& input_shape;3376 const float* input_data;3377 const RuntimeShape& output_shape;3378 float* output_data;3379 int start_batch;3380 int end_batch;