Function TEST

tensorflow/lite/kernels/shape_test.cc:54–61

};

TEST(ShapeOpTest, OutTypeInt) {
  ShapeOpModel<int32_t> model({1, 3, 1, 3, 5}, TensorType_FLOAT32,
                              TensorType_INT32);
  model.Invoke();

  EXPECT_THAT(model.GetOutput(), ElementsAreArray({1, 3, 1, 3, 5}));
  EXPECT_THAT(model.GetOutputShape(), ElementsAreArray({5}));
}

TEST(ShapeOpTest, OutTypeInt64) {
  ShapeOpModel<int64_t> model({1, 3, 1, 3, 5}, TensorType_FLOAT32,