Function TEST

tensorflow/lite/delegates/gpu/gl/kernels/conv_test.cc:33–69

namespace {

TEST(ConvTest, O2H2W1I1Stride1x1Dilation1x1) {
  TensorRef<BHWC> input;
  input.type = DataType::FLOAT32;
  input.ref = 0;
  input.shape = BHWC(1, 2, 2, 1);

  Convolution2DAttributes attr;
  Tensor<Linear, DataType::FLOAT32> bias;
  bias.shape.v = 2;
  bias.id = 1;
  bias.data = {1, 1};
  attr.bias = std::move(bias);

  Tensor<OHWI, DataType::FLOAT32> weights;
  weights.shape = OHWI(2, 2, 1, 1);
  weights.id = 2;
  weights.data = {1, 2, 3, 4};
  attr.weights = std::move(weights);

  attr.dilations = HW(1, 1);
  attr.padding.prepended = HW(0, 0);
  attr.padding.appended = HW(1, 0);
  attr.strides = HW(1, 1);

  TensorRef<BHWC> output;
  output.type = DataType::FLOAT32;
  output.ref = 3;
  output.shape = BHWC(1, 2, 2, 2);

  SingleOpModel model(
      {ToString(OperationType::CONVOLUTION_2D), std::move(attr)}, {input},
      {output});
  ASSERT_TRUE(model.PopulateTensor(0, {1, 1, 1, 1}));
  ASSERT_OK(model.Invoke(*NewConvolutionNodeShader()));
  EXPECT_THAT(model.GetOutput(0),
              Pointwise(FloatNear(1e-6), {4, 8, 4, 8, 2, 4, 2, 4}));
}

TEST(ConvTest, O1H2W2I1Stride1x1Dilation2x2) {
  TensorRef<BHWC> input;