ggml_conv_2d_sk_p0 function
- @Native<Pointer<
ggml_tensor> Function(Pointer<ggml_context> , Pointer<ggml_tensor> , Pointer<ggml_tensor> )>(ffi.Pointer<ggml_context>, ffi.Pointer<ggml_tensor>, ffi.Pointer<ggml_tensor>)>()
- 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.Native<
ffi.Pointer<ggml_tensor> Function(
ffi.Pointer<ggml_context>,
ffi.Pointer<ggml_tensor>,
ffi.Pointer<ggml_tensor>,
)
>()
external ffi.Pointer<ggml_tensor> ggml_conv_2d_sk_p0(
ffi.Pointer<ggml_context> ctx,
ffi.Pointer<ggml_tensor> a,
ffi.Pointer<ggml_tensor> b,
);