nn.EmbeddingBag is a frequently used layer in Recommendation models: https://github.com/facebookresearch/dlrm Would be great if the OpenCL backend can add support for this layer!
nn.EmbeddingBag is a frequently used layer in Recommendation models: https://github.com/facebookresearch/dlrm
Would be great if the OpenCL backend can add support for this layer!