Searched refs:adaptive_max_pool1d (Results 1 – 14 of 14) sorted by relevance
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | pooling.h | 376 return torch::adaptive_max_pool1d(input, output_size); in adaptive_max_pool1d_with_indices()
396 inline Tensor adaptive_max_pool1d( in adaptive_max_pool1d() function
417 inline Tensor adaptive_max_pool1d( in adaptive_max_pool1d() function
420 return detail::adaptive_max_pool1d(input, options.output_size()); in adaptive_max_pool1d()
|
| /external/pytorch/docs/source/ |
| D | nn.functional.rst | 44 adaptive_max_pool1d
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Pooling.cpp | 54 std::tuple<Tensor,Tensor> adaptive_max_pool1d(const Tensor & self, IntArrayRef output_size) { in adaptive_max_pool1d() function
|
| /external/pytorch/torch/ao/quantization/backend_config/ |
| D | utils.py | 158 elif p is F.adaptive_max_pool1d:
|
| /external/pytorch/torch/csrc/api/src/nn/modules/ |
| D | pooling.cpp | 149 return F::detail::adaptive_max_pool1d(input, options.output_size()); in forward()
|
| /external/pytorch/functorch/op_analysis/ |
| D | public_api | 444 nn.functional.adaptive_max_pool1d
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDecompositions.cpp | 56 OP_DECOMPOSE(adaptive_max_pool1d); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/nn/modules/ |
| D | pooling.py | 1292 return F.adaptive_max_pool1d(input, self.output_size, self.return_indices)
|
| /external/pytorch/test/cpp_api_parity/ |
| D | parity-tracker.md | 178 F::adaptive_max_pool1d|Yes|No
|
| /external/pytorch/torch/nn/ |
| D | functional.py | 1211 return torch.adaptive_max_pool1d(input, output_size)
1221 adaptive_max_pool1d,
1230 adaptive_max_pool1d = boolean_dispatch( variable
|
| /external/pytorch/torch/ |
| D | overrides.py | 434 torch.adaptive_max_pool1d: lambda inputs, output_size: -1,
827 … torch.nn.functional.adaptive_max_pool1d: lambda input, output_size, return_indices=False: -1,
|
| /external/pytorch/test/nn/ |
| D | test_pooling.py | 540 nn.functional.adaptive_max_pool1d,
|
| /external/pytorch/docs/source/notes/ |
| D | extending.rst | 843 'adaptive_max_pool1d', 'adaptive_max_pool1d_with_indices']
|
| /external/pytorch/test/cpp/api/ |
| D | functional.cpp | 480 auto y = F::adaptive_max_pool1d(x, F::AdaptiveMaxPool1dFuncOptions(3)); in TEST_F()
|