Function TEST

tensorflow/lite/kernels/log_softmax_test.cc:61–76

};

TEST(LogSoftmaxOpTest, SimpleTest) {
  LogSoftmaxOpModel m(/*batches=*/2, /*size=*/5);
  m.SetInput({
      1.0, 2.0, 3.0, 4.0, 5.0,       // b = 0
      -1.0, -2.0, -3.0, -4.0, -5.0,  // b = 1
  });

  m.Invoke();

  EXPECT_THAT(
      m.GetOutput(),
      ElementsAreArray(ArrayFloatNear(
          {-4.45191431, -3.45191431, -2.45191431, -1.45191443, -0.4519144,
           -0.4519144, -1.45191443, -2.45191431, -3.45191431, -4.45191431},
          1e-6)));
}

TEST(LogSoftmaxOpTest, CompareWithTFmini) {
  const int batch_size = 2;