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

Added missing enums.

Browse Source 
PiperOrigin-RevId: 329010576
Change-Id: Id408d6e7b4eddb45f3930de5f424b1b801e1cb29
This commit is contained in:
Raman Sarokin 2020-08-28 14:32:51 -07:00 committed by TensorFlower Gardener
parent 88eebe512c
commit 614f9f315b
1 changed files with 7 additions and 1 deletions

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

@ -798,6 +798,7 @@ class ElementwiseOperationParser : public TFLiteOperationParser {
case OperationType::ABS:
case OperationType::COPY:
case OperationType::COS:
case OperationType::ELU:
case OperationType::EXP:
case OperationType::LOG:
case OperationType::RSQRT:
@ -815,6 +816,8 @@ class ElementwiseOperationParser : public TFLiteOperationParser {
bool IsTwoArgumentOperation() const {
switch (operation_type_) {
case OperationType::DIV:
case OperationType::MAXIMUM:
case OperationType::MINIMUM:
case OperationType::POW:
case OperationType::SQUARED_DIFF:
case OperationType::SUB:
@ -826,8 +829,11 @@ class ElementwiseOperationParser : public TFLiteOperationParser {
bool IsTwoArgumentOperationWithConst() const {
switch (operation_type_) {
case OperationType::MINIMUM:
case OperationType::DIV:
case OperationType::MAXIMUM:
case OperationType::MINIMUM:
case OperationType::POW:
case OperationType::SQUARED_DIFF:
case OperationType::SUB:
return true;
default: