1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-12-29 15:11:58 +00:00
immich/machine-learning/ann
Fynn Petersen-Frey 753292956e
feat(ml): ARMNN acceleration (#5667)
* feat(ml): ARMNN acceleration for CLIP

* wrap ANN as ONNX-Session

* strict typing

* normalize ARMNN CLIP embedding

* mutex to handle concurrent execution

* make inputs contiguous

* fine-grained locking; concurrent network execution

---------

Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com>
2024-01-11 18:26:46 +01:00
..
export feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
__init__.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
ann.cpp feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
ann.py feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00
build.sh feat(ml): ARMNN acceleration (#5667) 2024-01-11 18:26:46 +01:00