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

Enable TransposeConv in hexagon delegate with version = 3 if the op doesn't have bias added.

Browse Source 
PiperOrigin-RevId: 348684103
Change-Id: I3c057e1535ac4df663c8ba0ae82c5ac7b4c61b4b
This commit is contained in:
Karim Nosir 2020-12-22 13:25:01 -08:00 committed by TensorFlower Gardener
parent d45ca04c30
commit 9d2d8ca23d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/delegates/hexagon/utils.cc
View File

@ -100,7 +100,6 @@ bool CheckOpVersion(const TfLiteRegistration* registration) {
case kTfLiteBuiltinSub: case kTfLiteBuiltinSub:
case kTfLiteBuiltinTanh: case kTfLiteBuiltinTanh:
case kTfLiteBuiltinTranspose: case kTfLiteBuiltinTranspose:
case kTfLiteBuiltinTransposeConv:
return registration->version <= 2; return registration->version <= 2;
case kTfLiteBuiltinRelu: case kTfLiteBuiltinRelu:
return registration->version == 2; return registration->version == 2;
@ -108,6 +107,7 @@ bool CheckOpVersion(const TfLiteRegistration* registration) {
case kTfLiteBuiltinDepthwiseConv2d: case kTfLiteBuiltinDepthwiseConv2d:
case kTfLiteBuiltinResizeBilinear: case kTfLiteBuiltinResizeBilinear:
case kTfLiteBuiltinResizeNearestNeighbor: case kTfLiteBuiltinResizeNearestNeighbor:
case kTfLiteBuiltinTransposeConv:
return registration->version <= 3; return registration->version <= 3;
case kTfLiteBuiltinFullyConnected: case kTfLiteBuiltinFullyConnected:
return registration->version <= 4; return registration->version <= 4;