| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Pooling.cpp | 43 Tensor adaptive_avg_pool1d(const Tensor & self, IntArrayRef output_size) { in adaptive_avg_pool1d() function
44 checkDimRange("adaptive_avg_pool1d", TensorArg(self, "self", 1), 2, 4 /* exclusive */); in adaptive_avg_pool1d()
45 check1d("adaptive_avg_pool1d", "output_size", output_size); in adaptive_avg_pool1d()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | pooling.h | 539 inline Tensor adaptive_avg_pool1d( in adaptive_avg_pool1d() function
542 return torch::adaptive_avg_pool1d(input, output_size); in adaptive_avg_pool1d()
548 /// https://pytorch.org/docs/main/nn.functional.html#torch.nn.functional.adaptive_avg_pool1d
558 /// F::adaptive_avg_pool1d(x, F::AdaptiveAvgPool1dFuncOptions(3));
560 inline Tensor adaptive_avg_pool1d( in adaptive_avg_pool1d() function
563 return detail::adaptive_avg_pool1d(input, options.output_size()); in adaptive_avg_pool1d()
|
| H A D | upsampling.h | 174 return detail::adaptive_avg_pool1d( in interpolate()
|
| /aosp_15_r20/external/pytorch/torch/ao/ns/fx/ |
| H A D | mappings.py | 72 F.adaptive_avg_pool1d,
543 torch.adaptive_avg_pool1d,
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | top_operators_github_usage.py | 377 ("nn.functional.adaptive_avg_pool1d", 43),
477 ("nn.AdaptiveAvgPool1d", 2481, "nn.functional.adaptive_avg_pool1d"),
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | pooling.h | 301 /// Options for `torch::nn::functional::adaptive_avg_pool1d`.
309 /// F::adaptive_avg_pool1d(x, F::AdaptiveAvgPool1dFuncOptions(3));
|
| /aosp_15_r20/external/pytorch/docs/source/scripts/ |
| H A D | build_quantization_configs.py | 49 # torch._VariableFunctionsClass.adaptive_avg_pool1d
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | helper.cpp | 129 "adaptive_avg_pool1d",
148 "adaptive_avg_pool1d",
|
| H A D | quantization_patterns.h | 827 auto adaptive_avg_pool1d = getInputTensorQParamOpFusionInfo( in quant_fusion_pattern_and_replacements() local
828 "aten::adaptive_avg_pool1d", {"%output_size"}); in quant_fusion_pattern_and_replacements()
1051 std::move(adaptive_avg_pool1d), in quant_fusion_pattern_and_replacements()
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | nn.functional.rst | 47 adaptive_avg_pool1d
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 441 nn.functional.adaptive_avg_pool1d
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 49 - aten::adaptive_avg_pool1d
732 aten::adaptive_avg_pool1d: 1
|
| H A D | model_ops.yaml | 47 aten::adaptive_avg_pool1d: 1
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/utils/ |
| H A D | op_registry.cpp | 31 "aten::adaptive_avg_pool1d(Tensor self, int[] output_size) -> Tensor", in nn_ops_first_input_preserving()
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_dtype_analysis.py | 39 "nn.functional.adaptive_avg_pool1d",
|
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
| H A D | parity-tracker.md | 181 F::adaptive_avg_pool1d|Yes|No
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesDecompositions.cpp | 57 OP_DECOMPOSE(adaptive_avg_pool1d); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | pooling.cpp | 191 return F::detail::adaptive_avg_pool1d(input, options.output_size()); in forward()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_script.cpp | 1263 def adaptive_avg_pool1d(self,
1269 return torch.adaptive_avg_pool1d(self, output_size), backward
|
| H A D | register_prim_ops_fulljit.cpp | 528 return at::adaptive_avg_pool1d( in interpolate()
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/backend_config/ |
| H A D | _common_operator_config_utils.py | 638 torch.adaptive_avg_pool1d,
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_dtensor_ops.py | 277 xfail("nn.functional.adaptive_avg_pool1d"),
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/ |
| H A D | utils.py | 61 torch.nn.functional.adaptive_avg_pool1d,
|
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_quantize_jit.py | 2900 self.adaptive_avg_pool1d = torch.nn.AdaptiveAvgPool1d(1)
2913 x = self.adaptive_avg_pool1d(x)
2919 x = F.adaptive_avg_pool1d(x, (1))
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | allowlist_for_publicAPI.json | 578 "adaptive_avg_pool1d",
|