diff options
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
| -rw-r--r-- | tensorflow/contrib/lite/model.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index 71e38c3f13..6c1ba3694a 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -616,6 +616,8 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type, break; } case BuiltinOperator_MEAN: + case BuiltinOperator_REDUCE_MAX: + case BuiltinOperator_REDUCE_PROD: case BuiltinOperator_SUM: { auto* params = MallocPOD<TfLiteReducerParams>(); if (auto* schema_params = op->builtin_options_as_ReducerOptions()) { |