Function TEST

tensorflow/lite/kernels/fill_test.cc:48–56

};

TEST(FillOpModel, FillInt32) {
  FillOpModel m({TensorType_INT32, {2}}, {TensorType_INT32});
  m.PopulateTensor<int32_t>(m.input1(), {2, 3});
  m.PopulateTensor<int32_t>(m.input2(), {-11});
  m.Invoke();
  EXPECT_THAT(m.ExtractVector<int32_t>(m.output()),
              ElementsAreArray({-11, -11, -11, -11, -11, -11}));
  EXPECT_THAT(m.GetTensorShape(m.output()), ElementsAreArray({2, 3}));
}

TEST(FillOpModel, FillInt64) {
  FillOpModel m({TensorType_INT32, {2}}, {TensorType_INT64});