Function create_image2d_from_tensor

src/core/CL/CLUtils.cpp:37–60

namespace arm_compute
{
cl::Image2D create_image2d_from_tensor(const ICLTensor *tensor, CLImage2DType image_type)
{
    ARM_COMPUTE_ERROR_ON_NULLPTR(tensor);

    const cl::Context &ctx    = CLKernelLibrary::get().context();
    const cl::Buffer  &buffer = tensor->cl_buffer();
    const ITensorInfo *info   = tensor->info();
    ARM_COMPUTE_ERROR_ON_MSG(info->lock_paddings(), "Tensor paddings must not be locked to allow extending paddings to "
                                                    "satisfy cl_image pitch alignment requirement");

    const size_t image_w{info->dimension(0) / 4};
    const size_t image_h{info->tensor_shape().total_size() / info->dimension(0)};
    const size_t max_image_w{CLKernelLibrary::get().get_device().getInfo<CL_DEVICE_IMAGE2D_MAX_WIDTH>()};
    const size_t max_image_h{CLKernelLibrary::get().get_device().getInfo<CL_DEVICE_IMAGE2D_MAX_HEIGHT>()};

    ARM_COMPUTE_UNUSED(max_image_w, max_image_h);
    ARM_COMPUTE_ERROR_ON_MSG(image_w > max_image_w, "Image width exceeds maximum width for exporting to cl_image");
    ARM_COMPUTE_ERROR_ON_MSG(image_h > max_image_h, "Image height exceeds maximum height for exporting to cl_image");

    const TensorShape shape2d(image_w, image_h);
    const size_t      image_row_pitch = info->strides_in_bytes()[1];

    return create_image2d_from_buffer(ctx, buffer, shape2d, info->data_type(), image_row_pitch, image_type);
}

cl::Image2D create_image2d_from_buffer(const cl::Context &ctx,
                                       const cl::Buffer  &buffer,