| 93 | }; |
| 94 | |
| 95 | TEST(ComparisonsTest, EqualBool) { |
| 96 | ComparisonOpModel model({1, 1, 1, 4}, {1, 1, 1, 4}, TensorType_BOOL, |
| 97 | BuiltinOperator_EQUAL); |
| 98 | model.PopulateTensor<bool>(model.input1(), {true, false, true, false}); |
| 99 | model.PopulateTensor<bool>(model.input2(), {true, true, false, false}); |
| 100 | model.Invoke(); |
| 101 | |
| 102 | EXPECT_THAT(model.GetOutput(), ElementsAre(true, false, false, true)); |
| 103 | EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 1, 1, 4)); |
| 104 | } |
| 105 | |
| 106 | TEST(ComparisonsTest, EqualFloat) { |
| 107 | ComparisonOpModel model({1, 1, 1, 4}, {1, 1, 1, 4}, TensorType_FLOAT32, |