diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-16 19:09:52 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 19:13:36 -0700 |
| commit | 1a2af489b1087eb22ec76863867e4e397e453e34 (patch) | |
| tree | d2576dbbfc6f77897e59353e0844aa0c75a23033 /tensorflow/contrib/lite/model.cc | |
| parent | d8f3425e5b054dff01b5ece80e8c8a101c4ed816 (diff) | |
Support reduce_max and reduce_prod
PiperOrigin-RevId: 204846139
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()) { |