| 57 | }; |
| 58 | |
| 59 | TEST(FloatSqueezeOpTest, SqueezeAll) { |
| 60 | std::initializer_list<float> data = { |
| 61 | 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, |
| 62 | 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0}; |
| 63 | FloatSqueezeOpModel m({TensorType_FLOAT32, {1, 24, 1}}, |
| 64 | {TensorType_FLOAT32, {24}}, {}); |
| 65 | m.SetInput(data); |
| 66 | m.Invoke(); |
| 67 | EXPECT_THAT(m.GetOutputShape(), ElementsAreArray({24})); |
| 68 | EXPECT_THAT( |
| 69 | m.GetOutput(), |
| 70 | ElementsAreArray({1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, |
| 71 | 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, |
| 72 | 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0})); |
| 73 | } |
| 74 | |
| 75 | TEST(FloatSqueezeOpTest, SqueezeSelectedAxis) { |
| 76 | std::initializer_list<float> data = { |
nothing calls this directly
no test coverage detected