| D | concatenation_test.cc | 103 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/1, in TEST() local
105 m0.SetInput(0, {1.0f, 3.0f, 4.0f, 7.0f}); in TEST()
106 m0.Invoke(); in TEST()
107 EXPECT_THAT(m0.GetOutput(), ElementsAreArray({1, 3, 4, 7})); in TEST()
111 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2, 1, 3}}, /*axis=*/2, in TEST() local
113 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, in TEST()
115 m0.Invoke(); in TEST()
116 EXPECT_THAT(m0.GetOutput(), in TEST()
121 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2, 1, 3}}, /*axis=*/0, in TEST() local
123 m0.SetInput(0, {1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, in TEST()
[all …]
|