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

make elementwise ops work

Browse Source
This commit is contained in:
Koan-Sin Tan 2019-05-11 09:42:21 +08:00
parent aa5110bdb2
commit 673939fd24
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");
} }