Function TEST_P

tensorflow/lite/kernels/reshape_test.cc:126–138

};

TEST_P(ReshapeOpTest, MismatchedDimensions) {
  if (GetParam() == kAsTensor) {
    ReshapeOpModel<float> m({1, 2, 4, 1}, {2}, {2, 1}, GetParam());
    m.SetInput({3});
    EXPECT_NE(m.InvokeUnchecked(), kTfLiteOk)
        << "num_input_elements != num_output_elements";
  } else {
#ifdef GTEST_HAS_DEATH_TEST
    EXPECT_DEATH(ReshapeOpModel<float>({1, 2, 4, 1}, {2}, {2, 1}, GetParam()),
                 "num_input_elements != num_output_elements");
#endif
  }
}

TEST_P(ReshapeOpTest, TooManyDimensions) {
#ifdef GTEST_HAS_DEATH_TEST