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.h143 struct MeanWorkerTask : cpu_backend_threadpool::Task { struct
144 MeanWorkerTask(const tflite::MeanParams& op_params, in MeanWorkerTask() function
160 void Run() override { in Run()
167 const RuntimeShape& input_shape;
168 const int8_t* input_data;
169 int32 multiplier;
170 int32 shift;
171 int32 bias;
172 const RuntimeShape& output_shape;
173 int8_t* output_data;
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1093 struct MeanWorkerTask : cpu_backend_threadpool::Task { struct
1094 MeanWorkerTask(const tflite::MeanParams& op_params, in MeanWorkerTask() argument
1110 void Run() override { in Run()
1117 const RuntimeShape& input_shape;
1118 const uint8_t* input_data;
1119 int32 multiplier;
1120 int32 shift;
1121 int32 bias;
1122 const RuntimeShape& output_shape;
1123 uint8_t* output_data;
[all …]