Home
last modified time | relevance | path

Searched refs:adaptive_max_pool2d_backward (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveMaxPooling2d.cpp56 TORCH_META_FUNC(adaptive_max_pool2d_backward) in TORCH_META_FUNC() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesPooling.cpp67 ALL_TENSORS_HAVE_OPTIONAL_BDIM_BOXED_CONTIG1(3, adaptive_max_pool2d_backward, 2); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect647 aten::adaptive_max_pool2d_backward
648 aten::adaptive_max_pool2d_backward.grad_input
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dgenerated_ops.cpp3803 aten::adaptive_max_pool2d_backward,
3813 p_node->Output(0) = at::cpu::adaptive_max_pool2d_backward( in __anon5d9c3eb916802()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py2904 @register_meta(aten.adaptive_max_pool2d_backward)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2286 make_fallback(aten.adaptive_max_pool2d_backward)
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_vmap.py5247 torch.ops.aten.adaptive_max_pool2d_backward(escaped, a, b)