diff options
author | 2018-09-18 16:39:12 -0700 | |
---|---|---|
committer | 2018-09-18 16:43:33 -0700 | |
commit | 86b8f034e4d3d3d12d1e9d1b94170b271491bed3 (patch) | |
tree | a602d74b9735ae8c50e539bc8fe81e9d971c09a2 /tensorflow/contrib/lite/delegates | |
parent | 073c418695ac9ef02071de3e08394e781ceca117 (diff) |
Reject RESHAPE if new_shape tensor is not provided.
PiperOrigin-RevId: 213541006
Diffstat (limited to 'tensorflow/contrib/lite/delegates')
-rw-r--r-- | tensorflow/contrib/lite/delegates/nnapi/nnapi_delegate.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/contrib/lite/delegates/nnapi/nnapi_delegate.cc index c6587b3d3f..d85e576284 100644 --- a/tensorflow/contrib/lite/delegates/nnapi/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/delegates/nnapi/nnapi_delegate.cc @@ -518,7 +518,7 @@ class NNAPIDelegateKernel { } break; case kTfLiteBuiltinReshape: - if (version == 1) { + if (version == 1 && node->inputs->size == 2) { return [](const NNAPIOpMappingArgs& mapping_args) -> ANeuralNetworksOperationType { return ANEURALNETWORKS_RESHAPE; |