Home
last modified time | relevance | path

Searched refs:max_pool3d_with_indices (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesPooling.cpp45 …elf, self_bdim, kernel_size, stride, padding, dilation, ceil_mode, 3, at::max_pool3d_with_indices); in max_pool3d_with_indices_batch_rule()
70 VMAP_SUPPORT(max_pool3d_with_indices, max_pool3d_with_indices_batch_rule); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/
H A Dpooling.h336 inline std::tuple<Tensor, Tensor> max_pool3d_with_indices( in max_pool3d_with_indices() function
343 return torch::max_pool3d_with_indices( in max_pool3d_with_indices()
357 inline std::tuple<Tensor, Tensor> max_pool3d_with_indices( in max_pool3d_with_indices() function
360 return detail::max_pool3d_with_indices( in max_pool3d_with_indices()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_autograd_functions.cpp23 call_fallback_fn<&ltc_eager_fallback, ATEN_OP(max_pool3d_with_indices)>:: in forward()
H A Dts_native_functions.cpp340 std::tuple<at::Tensor, at::Tensor> LazyNativeFunctions::max_pool3d_with_indices( in max_pool3d_with_indices() function in torch::lazy::LazyNativeFunctions
348 call_fallback_fn<&ltc_eager_fallback, ATEN_OP(max_pool3d_with_indices)>:: in max_pool3d_with_indices()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPooling.cpp182 auto output_and_indices = at::max_pool3d_with_indices( in max_pool3d()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dpooling.cpp133 return F::detail::max_pool3d_with_indices( in forward_with_indices()
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api429 max_pool3d_with_indices
/aosp_15_r20/external/pytorch/torch/nn/
H A Dfunctional.py844 def max_pool3d_with_indices( function
880 max_pool3d_with_indices,
892 return torch._C._nn.max_pool3d_with_indices(
927 if_true=max_pool3d_with_indices,
H A Dfunctional.pyi.in86 def max_pool3d_with_indices(
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect923 aten::max_pool3d_with_indices
924 aten::max_pool3d_with_indices.out
/aosp_15_r20/external/pytorch/test/
H A Dtest_namedtensor.py301 … check_tuple_return(F.max_pool3d_with_indices, [named_tensor_3d, [2, 2, 2]], named_tensor_3d.names)
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py1507 max_pool3d_with_indices = _onnx_symbolic("aten::max_pool3d_with_indices")( variable
1651 max_pool3d_with_indices,
/aosp_15_r20/external/pytorch/torch/
H A Doverrides.py932 torch.nn.functional.max_pool3d_with_indices: (
H A D_meta_registrations.py4419 @register_meta(aten.max_pool3d_with_indices)
/aosp_15_r20/external/pytorch/test/cpp/lazy/
H A Dtest_lazy_ops.cpp7554 auto outputs = torch::max_pool3d_with_indices( in TEST_F()
7563 auto lazy_outputs = torch::max_pool3d_with_indices( in TEST_F()
8104 std::tie(output, indices) = torch::max_pool3d_with_indices( in TEST_F()
10555 std::tie(output, indices) = torch::max_pool3d_with_indices( in TEST_F()
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py2229 make_fallback(aten.max_pool3d_with_indices) # @isuruf (can this one be implemented?)