Home
last modified time | relevance | path

Searched defs:MeanWorkerTask (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmean.h145 struct MeanWorkerTask : cpu_backend_threadpool::Task { struct
146 MeanWorkerTask(const tflite::MeanParams& op_params, in MeanWorkerTask() function
162 void Run() override { in Run()
169 const RuntimeShape& input_shape;
170 const int8_t* input_data;
171 int32 multiplier;
172 int32 shift;
173 int32 bias;
174 const RuntimeShape& output_shape;
175 int8_t* output_data;
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dreduce.h158 struct MeanWorkerTask : cpu_backend_threadpool::Task { struct
159 MeanWorkerTask(const tflite::MeanParams& op_params, in MeanWorkerTask() argument
175 void Run() override { in Run()
182 const RuntimeShape& input_shape;
183 const uint8_t* input_data;
184 int32 multiplier;
185 int32 shift;
186 int32 bias;
187 const RuntimeShape& output_shape;
188 uint8_t* output_data;
[all …]