Home
last modified time | relevance | path

Searched full:adaptive_avg_pool1d (Results 1 – 25 of 44) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/
DPooling.cpp43 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()
/external/pytorch/docs/source/scripts/
Dbuild_quantization_configs.py49 # torch._VariableFunctionsClass.adaptive_avg_pool1d
/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dpooling.h539 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()
Dupsampling.h174 return detail::adaptive_avg_pool1d( in interpolate()
/external/pytorch/docs/source/
Dnn.functional.rst47 adaptive_avg_pool1d
/external/pytorch/torch/ao/ns/fx/
Dmappings.py72 F.adaptive_avg_pool1d,
543 torch.adaptive_avg_pool1d,
/external/pytorch/functorch/op_analysis/
Dpublic_api441 nn.functional.adaptive_avg_pool1d
/external/pytorch/torch/csrc/jit/passes/utils/
Dop_registry.cpp31 "aten::adaptive_avg_pool1d(Tensor self, int[] output_size) -> Tensor", in nn_ops_first_input_preserving()
/external/pytorch/torch/csrc/api/include/torch/nn/options/
Dpooling.h301 /// Options for `torch::nn::functional::adaptive_avg_pool1d`.
309 /// F::adaptive_avg_pool1d(x, F::AdaptiveAvgPool1dFuncOptions(3));
/external/pytorch/torch/_functorch/
Dtop_operators_github_usage.py377 ("nn.functional.adaptive_avg_pool1d", 43),
477 ("nn.AdaptiveAvgPool1d", 2481, "nn.functional.adaptive_avg_pool1d"),
/external/pytorch/test/jit/
Dtest_dtype_analysis.py39 "nn.functional.adaptive_avg_pool1d",
/external/pytorch/torch/csrc/jit/passes/quantization/
Dhelper.cpp129 "adaptive_avg_pool1d",
148 "adaptive_avg_pool1d",
Dquantization_patterns.h827 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()
/external/pytorch/test/cpp_api_parity/
Dparity-tracker.md181 F::adaptive_avg_pool1d|Yes|No
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesDecompositions.cpp57 OP_DECOMPOSE(adaptive_avg_pool1d); in TORCH_LIBRARY_IMPL()
/external/pytorch/test/mobile/model_test/
Dmodel_ops.yaml47 aten::adaptive_avg_pool1d: 1
Dcoverage.yaml49 - aten::adaptive_avg_pool1d
732 aten::adaptive_avg_pool1d: 1
/external/pytorch/torch/csrc/api/src/nn/modules/
Dpooling.cpp191 return F::detail::adaptive_avg_pool1d(input, options.output_size()); in forward()
/external/pytorch/torch/ao/quantization/backend_config/
D_common_operator_config_utils.py638 torch.adaptive_avg_pool1d,
/external/pytorch/test/distributed/_tensor/
Dtest_dtensor_ops.py277 xfail("nn.functional.adaptive_avg_pool1d"),
/external/pytorch/torch/ao/quantization/
Dutils.py61 torch.nn.functional.adaptive_avg_pool1d,
/external/pytorch/torch/csrc/jit/runtime/
Dsymbolic_script.cpp1263 def adaptive_avg_pool1d(self,
1269 return torch.adaptive_avg_pool1d(self, output_size), backward
Dregister_prim_ops_fulljit.cpp528 return at::adaptive_avg_pool1d( in interpolate()
/external/pytorch/test/
Dallowlist_for_publicAPI.json578 "adaptive_avg_pool1d",
/external/pytorch/torch/testing/_internal/
Djit_metaprogramming_utils.py124 ('adaptive_avg_pool1d', (S, S, S), (5,), '', (True,)),

12