ggml_conv_2d_sk_p0 method
Pointer<ggml_tensor>
ggml_conv_2d_sk_p0(
- Pointer<
ggml_context> ctx, - Pointer<
ggml_tensor> a, - Pointer<
ggml_tensor> b
kernel size is a->ne0
x a->ne1
stride is equal to kernel size
padding is zero
example:
a: 16 16 3 768
b: 1024 1024 3 1
res: 64 64 768 1
used in sam
Implementation
ffi.Pointer<ggml_tensor> ggml_conv_2d_sk_p0(
ffi.Pointer<ggml_context> ctx,
ffi.Pointer<ggml_tensor> a,
ffi.Pointer<ggml_tensor> b,
) {
return _ggml_conv_2d_sk_p0(
ctx,
a,
b,
);
}