diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-18 16:39:12 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-18 16:43:33 -0700 |
| commit | 86b8f034e4d3d3d12d1e9d1b94170b271491bed3 (patch) | |
| tree | a602d74b9735ae8c50e539bc8fe81e9d971c09a2 /tensorflow/contrib/lite/nnapi_delegate.cc | |
| parent | 073c418695ac9ef02071de3e08394e781ceca117 (diff) | |
Reject RESHAPE if new_shape tensor is not provided.
PiperOrigin-RevId: 213541006
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
| -rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index a1c7434599..f23a0ccb80 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -512,6 +512,10 @@ TfLiteStatus AddOpsAndParams( nn_op_type = ANEURALNETWORKS_FULLY_CONNECTED; break; case tflite::BuiltinOperator_RESHAPE: + if (node.inputs->size != 2) { + logError("NNAPI only supports 2-input RESHAPE"); + return kTfLiteError; + } nn_op_type = ANEURALNETWORKS_RESHAPE; // add_reshape_params(node.builtin_data); break; |