ggml_conv_2d

Applies 2D convolution to input data.

Provides 'R' bindings to the 'GGML' tensor library for machine
learning, designed primarily for 'Vulkan' GPU acceleration with full CPU
fallback. 'Vulkan' support is auto-detected at build time on Linux (when
'libvulkan-dev' and 'glslc' are installed) and on Windows (when 'Vulkan'
'SDK' is installed and 'VULKAN_SDK' environment variable is set); all
operations fall back to CPU transparently when no GPU is available.
Implements tensor operations, neural network layers, quantization, and a
'Keras'-like sequential model API for building and training networks.
Includes 'AdamW' (Adam with Weight decay) and 'SGD' (Stochastic Gradient
Descent) optimizers with 'MSE' (Mean Squared Error) and cross-entropy
losses. Also provides a dynamic 'autograd' engine ('PyTorch'-style) with
data-parallel training via 'dp_train()', broadcast arithmetic, 'f16'
(half-precision) support on 'Vulkan' GPU, and a multi-head attention layer
for building Transformer architectures. Serves as backend for 'LLM' (Large
Language Model) inference via 'llamaR' and Stable Diffusion image
generation via 'sdR'. See <https://github.com/ggml-org/ggml> for more
information about the underlying library.

Yuri Baramykov

ggmlR

'GGML' Tensor Operations for Machine Learning

Georgi Gerganov

Jeffrey Quesnelle

Bowen Peng

Mozilla Foundation 

ggml_conv_2d function

<dl><dt>ctx</dt>
<dd>GGML context</dd>
<dt>a</dt>
<dd>Convolution kernel tensor [KW, KH, IC, OC]</dd>
<dt>b</dt>
<dd>Input data tensor [W, H, C, N]</dd>
<dt>s0</dt>
<dd>Stride dimension 0 (default 1)</dd>
<dt>s1</dt>
<dd>Stride dimension 1 (default 1)</dd>
<dt>p0</dt>
<dd>Padding dimension 0 (default 0)</dd>
<dt>p1</dt>
<dd>Padding dimension 1 (default 0)</dd>
<dt>d0</dt>
<dd>Dilation dimension 0 (default 1)</dd>
<dt>d1</dt>
<dd>Dilation dimension 1 (default 1)</dd></dl>

Arguments

2D Convolution (Graph) — ggml_conv_2d

<dl>

<dt>ctx</dt>
<dd>GGML context</dd>


<dt>a</dt>
<dd>Convolution kernel tensor [KW, KH, IC, OC]</dd>


<dt>b</dt>
<dd>Input data tensor [W, H, C, N]</dd>


<dt>s0</dt>
<dd>Stride dimension 0 (default 1)</dd>


<dt>s1</dt>
<dd>Stride dimension 1 (default 1)</dd>


<dt>p0</dt>
<dd>Padding dimension 0 (default 0)</dd>


<dt>p1</dt>
<dd>Padding dimension 1 (default 0)</dd>


<dt>d0</dt>
<dd>Dilation dimension 0 (default 1)</dd>


<dt>d1</dt>
<dd>Dilation dimension 1 (default 1)</dd>

</dl>

ggml_conv_2d: 2D Convolution (Graph)

Description

Usage

Value

Arguments