Home
last modified time | relevance | path

Searched refs:adaptive_max_pool3d_backward (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesPooling.cpp68 ALL_TENSORS_HAVE_OPTIONAL_BDIM_BOXED_CONTIG1(4, adaptive_max_pool3d_backward, 2); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveMaxPooling3d.cpp67 TORCH_META_FUNC(adaptive_max_pool3d_backward) in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect651 aten::adaptive_max_pool3d_backward
652 aten::adaptive_max_pool3d_backward.grad_input
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dgenerated_ops.cpp3828 aten::adaptive_max_pool3d_backward,
3838 p_node->Output(0) = at::cpu::adaptive_max_pool3d_backward( in __anon5d9c3eb916a02()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py2968 @register_meta(aten.adaptive_max_pool3d_backward)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2287 make_fallback(aten.adaptive_max_pool3d_backward)