Searched defs:embedding_bag (Results 1 – 8 of 8) sorted by relevance
89 inline Tensor embedding_bag( in embedding_bag() function
229 def embedding_bag( function
590 def embedding_bag( function
1203 def embedding_bag( function
931 def embedding_bag( function
580 def embedding_bag(fake_mode, func, *args, **kwargs): function
1227 embedding_bag(const Tensor &weight, const Tensor &indices, in embedding_bag() function1259 embedding_bag(const Tensor &weight, const Tensor &indices, in embedding_bag() function
2554 def embedding_bag( function