| 260 | } |
| 261 | |
| 262 | void ResolveAxis(const int* axis_data, int axis_count, |
| 263 | tflite::MeanParams* op_params) { |
| 264 | int i = 0; |
| 265 | for (; i < axis_count; ++i) { |
| 266 | op_params->axis[i] = static_cast<int16>(axis_data[i]); |
| 267 | } |
| 268 | for (; i < 4; ++i) { |
| 269 | op_params->axis[i] = 1; |
| 270 | } |
| 271 | } |
| 272 | |
| 273 | template <KernelType kernel_type> |
| 274 | TfLiteStatus EvalMean(TfLiteContext* context, TfLiteNode* node) { |