Function TEST

tensorflow/lite/kernels/rank_test.cc:52–58

};

TEST(RankOpTest, InputTypeFloat) {
  RankOpModel model({1, 3, 1, 3, 5}, TensorType_FLOAT32);
  model.Invoke();

  EXPECT_THAT(model.GetOutput(), ElementsAreArray({5}));
  EXPECT_TRUE(model.GetOutputShape().empty());
}

TEST(RankOpTest, InputTypeInt) {
  RankOpModel model({1, 3, 1, 3, 5}, TensorType_INT32);