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

Fix condition for activation function. FullyConnected in hexagon delegate doesn't support activation currently.

Browse Source 
PiperOrigin-RevId: 312292018
Change-Id: I921f00a42d1092f8910acf563cd0e97afd46ddc2
This commit is contained in:
Karim Nosir 2020-05-19 09:12:46 -07:00 committed by TensorFlower Gardener
parent e25d9862ca
commit 282db86128
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/experimental/delegates/hexagon/utils.cc
View File

@ -212,7 +212,7 @@ bool IsNodeSupportedByHexagon(const TfLiteRegistration* registration,
reinterpret_cast<const TfLiteFullyConnectedParams*>(
node->builtin_data);
return (weights_const && bias_const_or_no_bias &&
IsActivationReluOrNone(matmul_params->activation) &&
matmul_params->activation == kTfLiteActNone &&
matmul_params->keep_num_dims == false &&
matmul_params->weights_format ==
kTfLiteFullyConnectedWeightsFormatDefault);