diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-17 06:51:58 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-17 06:55:33 -0700 |
| commit | 8f255771c0ead16149fb003a9da45ff7346159d3 (patch) | |
| tree | 4e9ba5c66474e7994622b5116257ea5b3aeef42d /tensorflow/contrib/lite/nnapi_delegate.cc | |
| parent | 5cb77a7ac4741df72e1739c4fda3f552afc9c47c (diff) | |
Implement reduce_sum
PiperOrigin-RevId: 200895985
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
| -rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index 999c31d4bf..8d506f562f 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -312,7 +312,7 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, }; auto add_mean_params = [&add_scalar_int32](void* data) { - auto builtin = reinterpret_cast<TfLiteMeanParams*>(data); + auto builtin = reinterpret_cast<TfLiteReducerParams*>(data); add_scalar_int32(builtin->keep_dims); }; @@ -500,6 +500,7 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, case tflite::BuiltinOperator_SPARSE_TO_DENSE: case tflite::BuiltinOperator_EQUAL: case tflite::BuiltinOperator_NOT_EQUAL: + case tflite::BuiltinOperator_SUM: FATAL("Op code %d is currently not delegated to NNAPI", builtin); nn_op_type = -1; // set to invalid break; |