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

Merge pull request #28618 from freedomtan:fix_elmementwise_op_problem_in_gpu_delegate

Browse Source 
PiperOrigin-RevId: 247950785
This commit is contained in:
TensorFlower Gardener 2019-05-13 09:50:15 -07:00
commit fdd04bd83d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/delegates/gpu/common/model_builder.cc
View File

@ -1240,8 +1240,7 @@ class ElementwiseOperationParser : public TFLiteOperationParser {
TfLiteSubParams* tf_options = nullptr; TfLiteSubParams* tf_options = nullptr;
RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options)); RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &tf_options));
RETURN_IF_ERROR(CheckActivationSupported(tf_options->activation)); RETURN_IF_ERROR(CheckActivationSupported(tf_options->activation));
} } else if (!IsOneArgumentOperation()) {
if (!IsOneArgumentOperation()) {
return InvalidArgumentError("Incorrect operation type passed"); return InvalidArgumentError("Incorrect operation type passed");
} }