| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesPooling.cpp | 45 …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 D | pooling.h | 336 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 D | ts_autograd_functions.cpp | 23 call_fallback_fn<<c_eager_fallback, ATEN_OP(max_pool3d_with_indices)>:: in forward()
|
| H A D | ts_native_functions.cpp | 340 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<<c_eager_fallback, ATEN_OP(max_pool3d_with_indices)>:: in max_pool3d_with_indices()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Pooling.cpp | 182 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 D | pooling.cpp | 133 return F::detail::max_pool3d_with_indices( in forward_with_indices()
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 429 max_pool3d_with_indices
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | functional.py | 844 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 D | functional.pyi.in | 86 def max_pool3d_with_indices(
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_has_decomposition.expect | 923 aten::max_pool3d_with_indices
924 aten::max_pool3d_with_indices.out
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_namedtensor.py | 301 … 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 D | symbolic_opset9.py | 1507 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 D | overrides.py | 932 torch.nn.functional.max_pool3d_with_indices: (
|
| H A D | _meta_registrations.py | 4419 @register_meta(aten.max_pool3d_with_indices)
|
| /aosp_15_r20/external/pytorch/test/cpp/lazy/ |
| H A D | test_lazy_ops.cpp | 7554 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 D | lowering.py | 2229 make_fallback(aten.max_pool3d_with_indices) # @isuruf (can this one be implemented?)
|