Function TEST

tensorflow/lite/delegates/gpu/gl/kernels/mul_test.cc:33–51

namespace {

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

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

  MultiplyScalarAttributes attr;
  attr.param = 2;

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

TEST(MulTest, Linear) {
  TensorRef<BHWC> input;