Function TEST_F

tensorflow/lite/delegates/gpu/cl/kernels/apply_mask_test.cc:34–61

namespace {

TEST_F(OpenCLOperationTest, ApplyMaskOneChannel) {
  TensorFloat32 src_tensor;
  src_tensor.shape = BHWC(1, 2, 2, 2);
  src_tensor.data = {-4.0f, -3.0f, -1.0f, 0.0f, 1.0f, 3.0f, 4.0f, 6.0f};
  TensorFloat32 mask_tensor;
  mask_tensor.shape = BHWC(1, 2, 2, 1);
  mask_tensor.data = {2.0f, 0.5f, 1.0f, 0.0f};

  for (auto storage : env_.GetSupportedStorages()) {
    for (auto precision : env_.GetSupportedPrecisions()) {
      const float eps = precision == CalculationsPrecision::F32 ? 1e-6f : 1e-3f;
      OperationDef op_def;
      op_def.precision = precision;
      auto data_type = DeduceDataTypeFromPrecision(precision);
      op_def.src_tensors.push_back({data_type, storage});
      op_def.src_tensors.push_back({data_type, storage});
      op_def.dst_tensors.push_back({data_type, storage});
      TensorFloat32 dst_tensor;
      ApplyMask operation = CreateApplyMask(op_def);
      ASSERT_OK(ExecuteGPUOperation({src_tensor, mask_tensor},
                                    creation_context_, &operation,
                                    BHWC(1, 2, 2, 2), &dst_tensor));
      EXPECT_THAT(dst_tensor.data,
                  Pointwise(FloatNear(eps), {-8.0f, -6.0f, -0.5f, 0.0f, 1.0f,
                                             3.0f, 0.0f, 0.0f}));
    }
  }
}

TEST_F(OpenCLOperationTest, ApplyMaskEqualSizes) {
  TensorFloat32 src_tensor;