diff options
author | 2018-06-17 06:51:58 -0700 | |
---|---|---|
committer | 2018-06-17 06:55:33 -0700 | |
commit | 8f255771c0ead16149fb003a9da45ff7346159d3 (patch) | |
tree | 4e9ba5c66474e7994622b5116257ea5b3aeef42d /tensorflow/contrib/lite/model.cc | |
parent | 5cb77a7ac4741df72e1739c4fda3f552afc9c47c (diff) |
Implement reduce_sum
PiperOrigin-RevId: 200895985
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
-rw-r--r-- | tensorflow/contrib/lite/model.cc | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index bc62e4cc2d..b9d100b7c9 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -597,9 +597,10 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type, *builtin_data = reinterpret_cast<void*>(params); break; } - case BuiltinOperator_MEAN: { - auto* params = MallocPOD<TfLiteMeanParams>(); - if (auto* schema_params = op->builtin_options_as_MeanOptions()) { + case BuiltinOperator_MEAN: + case BuiltinOperator_SUM: { + auto* params = MallocPOD<TfLiteReducerParams>(); + if (auto* schema_params = op->builtin_options_as_ReducerOptions()) { params->keep_dims = schema_params->keep_dims(); } *builtin_data = reinterpret_cast<void*>(params); |