| 51 | }; |
| 52 | |
| 53 | TEST(PowOpModel, Simple) { |
| 54 | PowOpModel<int32_t> model({TensorType_INT32, {1, 2, 2, 1}}, |
| 55 | {TensorType_INT32, {1, 2, 2, 1}}, |
| 56 | {TensorType_INT32, {}}); |
| 57 | model.PopulateTensor<int32_t>(model.input1(), {12, 2, 7, 8}); |
| 58 | model.PopulateTensor<int32_t>(model.input2(), {1, 2, 3, 1}); |
| 59 | model.Invoke(); |
| 60 | EXPECT_THAT(model.GetOutputShape(), ElementsAre(1, 2, 2, 1)); |
| 61 | EXPECT_THAT(model.GetOutput(), ElementsAre(12, 4, 343, 8)); |
| 62 | } |
| 63 | |
| 64 | TEST(PowOpModel, NegativeAndZeroValue) { |
| 65 | PowOpModel<int32_t> model({TensorType_INT32, {1, 2, 2, 1}}, |
nothing calls this directly
no test coverage detected