experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tflite] bump SPLIT op ver from 1 to 3 in NNAPI delegate

Browse Source 
I need SPLIT op version 3. Since it's supported by TFLite and
NNAPI 1.2. It's should be safe to bump the op version so that
I can delegate SPLIT ops to accelerators.
This commit is contained in:
Koan-Sin Tan 2019-12-05 21:26:27 +08:00
parent 5aa0256315
commit ae2e9865a1
1 changed files with 1 additions and 1 deletions

2
tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
View File

@ -2023,7 +2023,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