[hannk] Fix > and >= op implementations (halide#6312)

a>b should be b<=a (not b <a) a>=b should be b<a (not b<=a)
xiebinbb · Oct 12, 2021 · d4e45bd · d4e45bd
1 parent 89b36b4
commit d4e45bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/apps/hannk/delegate/hannk_delegate.cpp b/apps/hannk/delegate/hannk_delegate.cpp
@@ -583,11 +583,11 @@ class HannkDelegateKernel final {
     }
 
     OpPtr BuildGreater(TfLiteContext *context, TfLiteNode *node) {
-        return BuildBinary(context, node, BinaryOp::Less, true);
+        return BuildBinary(context, node, BinaryOp::LessEqual, true);
     }
 
     OpPtr BuildGreaterEqual(TfLiteContext *context, TfLiteNode *node) {
-        return BuildBinary(context, node, BinaryOp::LessEqual, true);
+        return BuildBinary(context, node, BinaryOp::Less, true);
     }
 
     OpPtr BuildEqual(TfLiteContext *context, TfLiteNode *node) {

diff --git a/apps/hannk/tflite/tflite_parser.cpp b/apps/hannk/tflite/tflite_parser.cpp
@@ -438,9 +438,9 @@ class Parser {
         case tflite::BuiltinOperator_GATHER:
             return parse_gather(op);
         case tflite::BuiltinOperator_GREATER:
-            return parse_binary(op, BinaryOp::Less, true);
-        case tflite::BuiltinOperator_GREATER_EQUAL:
             return parse_binary(op, BinaryOp::LessEqual, true);
+        case tflite::BuiltinOperator_GREATER_EQUAL:
+            return parse_binary(op, BinaryOp::Less, true);
         case tflite::BuiltinOperator_L2_NORMALIZATION:
             return parse_l2_normalization(op);
         case tflite::BuiltinOperator_LESS: