diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc index c477125a49a..d09c1dc5b33 100644 --- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc +++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc @@ -2195,7 +2195,7 @@ bool NNAPIDelegateKernel::Validate( "NNAPI only supports constant int32 axis tensor.", &val_ctx); } break; case kTfLiteBuiltinSplit: { - ExpectOpVersion(version, 1, &val_ctx); + ExpectOpVersion(version, 3, &val_ctx); ExpectMinAndroidSdkVersion(android_sdk_version, kMinSdkVersionForNNAPI12, &val_ctx); // Tensor indices: split_dim: 0, value: 1