Embedding: Correct the prams when create `bag_size` tensor - Githubissues

intel / torch-xpu-ops

Apache License 2.0

30 stars 21 forks source link

Embedding: Correct the prams when create `bag_size` tensor #1018

Closed chunhuanMeng closed 1 week ago

chunhuanMeng commented 4 weeks ago

Resolve issue https://github.com/intel/torch-xpu-ops/issues/774