starred/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2026-04-20 15:55:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Michael Yang
2025-01-22 10:51:42 -08:00
parent 144f63e2fb
commit 8ab13e4d3e
59 changed files with 4001 additions and 857 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
ml/nn/convolution.go Normal file
View File

@@ -0,0 +1,11 @@
package nn
import "github.com/ollama/ollama/ml"
type Conv2D struct {
Weight ml.Tensor `gguf:"weight"`
}
func (m *Conv2D) Forward(ctx ml.Context, t ml.Tensor, s0, s1, p0, p1, d0, d1 int) ml.Tensor {
return m.Weight.Conv2D(ctx, t, s0, s1, p0, p1, d0, d1)
}