Searched refs:adaptive_max_pool2d (Results 1 – 20 of 20) sorted by relevance
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | pooling.h | 430 return torch::adaptive_max_pool2d(input, output_size_); in adaptive_max_pool2d_with_indices()
452 inline Tensor adaptive_max_pool2d( in adaptive_max_pool2d() function
473 inline Tensor adaptive_max_pool2d( in adaptive_max_pool2d() function
476 return detail::adaptive_max_pool2d(input, options.output_size()); in adaptive_max_pool2d()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | AdaptiveMaxPooling2d.cpp | 15 TORCH_META_FUNC(adaptive_max_pool2d) (const Tensor& input, IntArrayRef output_size) { in TORCH_META_FUNC() argument
|
| D | Pooling.cpp | 71 auto [output, indices] = at::adaptive_max_pool2d( in adaptive_max_pool1d()
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesPooling.cpp | 65 EXISTING_BDIM_ALL_BOXED(adaptive_max_pool2d); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/docs/source/ |
| D | nn.functional.rst | 45 adaptive_max_pool2d
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | utils.py | 160 elif p is F.adaptive_max_pool2d:
|
| /external/pytorch/aten/src/ATen/native/mps/operations/ |
| D | AdaptivePooling.mm | 186 … "adaptive_max_pool2d(): Expected input to have non-zero size for non-batch dimensions, "
|
| /external/pytorch/torch/csrc/api/src/nn/modules/ |
| D | pooling.cpp | 159 return F::detail::adaptive_max_pool2d(input, options.output_size()); in forward()
|
| /external/pytorch/functorch/op_analysis/ |
| D | public_api | 445 nn.functional.adaptive_max_pool2d
|
| /external/pytorch/torch/nn/modules/ |
| D | pooling.py | 1334 return F.adaptive_max_pool2d(input, self.output_size, self.return_indices)
|
| /external/pytorch/test/cpp_api_parity/ |
| D | parity-tracker.md | 179 F::adaptive_max_pool2d|Yes|No
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 645 aten::adaptive_max_pool2d
646 aten::adaptive_max_pool2d.out
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 1267 return torch._C._nn.adaptive_max_pool2d(input, output_size)
1277 adaptive_max_pool2d,
1286 adaptive_max_pool2d = boolean_dispatch( variable
|
| /external/pytorch/test/nn/ |
| D | test_pooling.py | 530 nn.functional.adaptive_max_pool2d,
1945 imgs_ = F.adaptive_max_pool2d(imgs, (Oh, Ow))
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 4284 aten.adaptive_max_pool2d.default, add_to_fallback_set=False
4288 @register_lowering(aten.adaptive_max_pool2d)
4289 def adaptive_max_pool2d(x, output_size): function
|
| /external/pytorch/torch/ |
| D | overrides.py | 829 … torch.nn.functional.adaptive_max_pool2d: lambda input, output_size, return_indices=False: -1,
|
| D | _meta_registrations.py | 2856 @register_meta(aten.adaptive_max_pool2d)
|
| /external/pytorch/test/cpp/api/ |
| D | functional.cpp | 489 auto y = F::adaptive_max_pool2d(x, F::AdaptiveMaxPool2dFuncOptions(3)); in TEST_F()
|
| /external/pytorch/test/inductor/ |
| D | test_torchinductor.py | 3799 return aten.adaptive_max_pool2d(x, (6, 6))
3822 return aten.adaptive_max_pool2d(x, (4, 4))
|
| /external/pytorch/test/functorch/ |
| D | test_vmap.py | 5235 torch.nn.functional.adaptive_max_pool2d(escaped, output_size=(1, 1))
|