Home
last modified time | relevance | path

Searched defs:embedding_bag (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dembedding.h89 inline Tensor embedding_bag( in embedding_bag() function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset18.py229 def embedding_bag( function
H A Dsymbolic_opset10.py590 def embedding_bag( function
H A Dsymbolic_opset11.py1203 def embedding_bag( function
H A Dsymbolic_opset9.py931 def embedding_bag( function
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py580 def embedding_bag(fake_mode, func, *args, **kwargs): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DEmbeddingBag.cpp1227 embedding_bag(const Tensor &weight, const Tensor &indices, in embedding_bag() function
1259 embedding_bag(const Tensor &weight, const Tensor &indices, in embedding_bag() function
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py2554 def embedding_bag( function