Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_autograd_functions.cpp43 ATEN_OP(max_pool3d_with_indices_backward)>:: in backward()
H A Dts_native_functions.cpp352 at::Tensor LazyNativeFunctions::max_pool3d_with_indices_backward( in max_pool3d_with_indices_backward() function in torch::lazy::LazyNativeFunctions
363 ATEN_OP(max_pool3d_with_indices_backward)>:: in max_pool3d_with_indices_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesPooling.cpp72 ALL_TENSORS_HAVE_OPTIONAL_BDIM_BOXED_CONTIG1(4, max_pool3d_with_indices_backward, 2); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect925 aten::max_pool3d_with_indices_backward
926 aten::max_pool3d_with_indices_backward.grad_input
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py4524 @register_meta(aten.max_pool3d_with_indices_backward)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2283 make_fallback(aten.max_pool3d_with_indices_backward)