Skip to content

Commit 1991fad

Browse files
Change the test case of "unsupported" OP from add to floor_mod. Do this since add OP will be supported later, but this test case can remain. Using an OP QNN does not support instead.
PiperOrigin-RevId: 683416587
1 parent 768e03c commit 1991fad

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

tensorflow/lite/experimental/lrt/qnn/IR/qnn_op_test.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ TEST(TestLegalizeOp, SimpleSupportedOp) {
4949
}
5050

5151
TEST(TestLegalizeOp, UnsupportedOp) {
52-
auto model = LoadTestFileModel("add_simple.tflite");
52+
auto model = LoadTestFileModel("simple_floor_mod_op.tflite");
5353
ASSERT_RESULT_OK_ASSIGN(auto subgraph,
5454
::graph_tools::GetSubgraph(model.get()));
5555
ASSERT_RESULT_OK_ASSIGN(auto ops, ::graph_tools::GetSubgraphOps(subgraph));
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
module {
2+
func.func @main(%arg0: tensor<5xf32>, %arg1: tensor<5xf32>) -> tensor<5xf32> {
3+
%0 = "tfl.floor_mod"(%arg0, %arg1) : (tensor<5xf32>, tensor<5xf32>) -> tensor<5xf32>
4+
return %0 : tensor<5xf32>
5+
}
6+
}

0 commit comments

Comments
 (0)