Function TEST

tensorflow/lite/kernels/embedding_lookup_sparse_test.cc:87–100

};

TEST(EmbeddingLookupOpTest, SimpleTest) {
  EmbeddingLookupSparseOpModel m(CombinerType_SUM, {3}, {3, 2}, {2}, {4, 3, 2});
  m.SetInput({1, 3, 0}, {0, 0, 2, 0, 2, 1}, {3, 2}, {1.0, 2.0, 4.0});
  m.Set3DWeightMatrix(
      [](int i, int j, int k) { return i + j / 10.0f + k / 100.0f; });
  m.Invoke();

  EXPECT_THAT(m.GetOutput(),
              ElementsAreArray(ArrayFloatNear({
                  1.00, 1.01, 1.10, 1.11, 1.20, 1.21,  // Row 1
                  0.00, 0.00, 0.00, 0.00, 0.00, 0.00,  // -
                  6.00, 6.06, 6.60, 6.66, 7.20, 7.26,  // 2 * Row 3 + 4 * Row 0
              })));
}

TEST(EmbeddingLookupOpTest, SimpleTestMean) {
  EmbeddingLookupSparseOpModel m(CombinerType_MEAN, {3}, {3, 2}, {2},