Home
last modified time | relevance | path

Searched refs:_embedding_bag_backward (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNonSymbolicBC.h19 TORCH_API at::Tensor _embedding_bag_backward(const at::Tensor & grad, const at::Tensor & indices, c…
H A DEmbeddingBag.cpp1374 Tensor _embedding_bag_backward(const Tensor &grad, const Tensor &indices_, in _embedding_bag_backward() function
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect99 aten::_embedding_bag_backward
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py6348 @register_meta(aten._embedding_bag_backward)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2275 make_fallback(aten._embedding_bag_backward)