Function TEST_F

tensorflow/lite/delegates/gpu/cl/kernels/concat_test.cc:34–65

namespace {

TEST_F(OpenCLOperationTest, ConcatWidth) {
  TensorFloat32 src0, src1;
  src0.shape = BHWC(1, 2, 1, 2);
  src0.data = {half(0.0f), half(-1.0f), half(-0.05f), half(0.045f)};
  src1.shape = BHWC(1, 2, 2, 2);
  src1.data = {half(1.0f), half(-1.2f), half(-0.45f), half(1.045f),
               half(1.1f), half(-1.3f), half(-0.55f), half(2.045f)};

  ConcatAttributes attr;
  attr.axis = Axis::WIDTH;

  for (auto storage : env_.GetSupportedStorages()) {
    for (auto precision : env_.GetSupportedPrecisions()) {
      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;
      ConcatXY operation = CreateConcatXY(op_def, attr, 2);
      ASSERT_OK(ExecuteGPUOperation({src0, src1}, creation_context_, &operation,
                                    BHWC(1, 2, 3, 2), &dst_tensor));
      EXPECT_THAT(
          dst_tensor.data,
          Pointwise(FloatNear(0.0f),
                    {half(0.0f), half(-1.0f), half(1.0f), half(-1.2f),
                     half(-0.45f), half(1.045f), half(-0.05f), half(0.045f),
                     half(1.1f), half(-1.3f), half(-0.55f), half(2.045f)}));
    }
  }
}

TEST_F(OpenCLOperationTest, ConcatHeight) {
  TensorFloat32 src0, src1;