[Opset15] Implementation for EmbeddingBagPacked and EmbeddingBagOffsets ops

openvinotoolkit / openvino

OpenVINO™ is an open-source toolkit for optimizing and deploying AI inference

https://docs.openvino.ai

Apache License 2.0

6.12k stars 2.04k forks source link

Closed mmikolajcz closed 1 week ago

mmikolajcz commented 2 weeks ago

Create PoC implementation for EmbeddingBag Offsets and Packed that introduce mode 'mean'
Add Core, Reference and CPU implementation for both ops

Specification will be added in separate PR