]> git.djapps.eu Git - pkg/ggml/sources/ggml/commit
ggml: add support for float16 input tensors in pooling operations (#895)
authorIvan Filipov <redacted>
Mon, 22 Jul 2024 11:32:02 +0000 (14:32 +0300)
committerGitHub <redacted>
Mon, 22 Jul 2024 11:32:02 +0000 (14:32 +0300)
commit2560b8ddc6c106c1334b5d6350632718e1c87590
treeb8ec54ef4dc0215f0224794fcca93570a304e3e2
parentf18a111977d3f6730c2df977e846701f57bb4d15
ggml: add support for float16 input tensors in pooling operations (#895)

* Add support for float16 tensors in 1d pooling operations

* Add support for float16 input tensors in 2d pooling operations

* code cleanup

remove unnecessary casting during srow ptr initialization

---------

Co-authored-by: vanaka11 <redacted>
src/ggml.c
tests/test-pool.c