| D | reshape_test.cc | 47 class ReshapeOpModel : public SingleOpModel { class
49 ReshapeOpModel(std::initializer_list<int> input_shape, in ReshapeOpModel() function in tflite::__anondc2713f00111::ReshapeOpModel
129 ReshapeOpModel<float> m({1, 2, 4, 1}, {2}, {2, 1}, GetParam()); in TEST_P()
133 EXPECT_DEATH(ReshapeOpModel<float>({1, 2, 4, 1}, {2}, {2, 1}, GetParam()), in TEST_P()
141 EXPECT_DEATH(ReshapeOpModel<float>({1, 1, 2, 1, 1, 1, 1, 1, 1}, {9}, in TEST_P()
151 ReshapeOpModel<float>({1, 2, 4, 1}, {4}, {-1, -1, 2, 4}, GetParam()), in TEST_P()
154 ReshapeOpModel<float> m({1, 2, 4, 1}, {4}, {-1, -1, 2, 4}, GetParam()); in TEST_P()
163 ReshapeOpModel<float> m({1, 2, 2}, {2, 2}, {1, 2, 2, 1}, GetParam()); in TEST_P()
172 ReshapeOpModel<float> m({1, 2, 4, 1}, {3}, {2, 2, 2}, GetParam()); in TEST_P()
180 ReshapeOpModel<float> m({1, 2, 4, 1}, {3}, {2, 1, -1}, GetParam()); in TEST_P()
[all …]
|