| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Pooling.cpp | 58 bool count_include_pad, in mkldnn_avg_pool2d()
68 bool count_include_pad, in mkldnn_avg_pool2d_out()
80 bool count_include_pad, in mkldnn_avg_pool3d()
90 bool count_include_pad, in mkldnn_avg_pool3d_out()
142 bool count_include_pad, in mkldnn_avg_pool2d_backward_out()
155 bool count_include_pad, in mkldnn_avg_pool2d_backward()
166 bool count_include_pad, in mkldnn_avg_pool3d_backward_out()
179 bool count_include_pad, in mkldnn_avg_pool3d_backward()
420 bool count_include_pad, in mkldnn_avg_pool2d()
445 bool count_include_pad, in mkldnn_avg_pool2d_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AvgPoolKernel.cpp | 23 bool count_include_pad, in cpu_avg_pool2d()
110 bool count_include_pad, in cpu_avg_pool2d_channels_last()
224 bool count_include_pad, in cpu_avg_pool2d_channels_last()
355 bool count_include_pad, in cpu_avg_pool2d_backward()
423 bool count_include_pad, in cpu_avg_pool2d_backward_channels_last()
502 bool count_include_pad, in avg_pool2d_kernel_impl()
528 bool count_include_pad, in avg_pool2d_backward_kernel_impl()
556 bool count_include_pad, in cpu_avg_pool3d()
652 bool count_include_pad, in cpu_avg_pool3d_channels_last()
775 bool count_include_pad, in cpu_avg_pool3d_channels_last()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_avg_pool2d_test.cpp | 28 bool count_include_pad, in op_avg_pool2d_out()
107 bool count_include_pad = true; in TEST_F() local
259 bool count_include_pad = true; in TEST_F() local
490 bool count_include_pad = false; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AveragePool2d.cpp | 49 bool count_include_pad, in avg_pool2d_out_frame()
187 bool count_include_pad, in q_avg_pool2d()
267 bool count_include_pad, in qnnpack_avg_pool2d()
364 bool count_include_pad, in avg_pool2d_quantized_cpu()
|
| H A D | AveragePool3d.cpp | 102 bool count_include_pad, in q_avg_pool3d()
167 bool count_include_pad, in avg_pool3d_quantized_cpu()
|
| H A D | AdaptiveAveragePooling.cpp | 274 bool count_include_pad{false}; in qnnpack_adaptive_avg_pool2d() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Pool.cpp | 127 bool count_include_pad; member
133 const ValueRef count_include_pad) { in create_divisor_params()
148 const ValueRef count_include_pad, in add_avg_pool2d_node()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 39 const bool count_include_pad, const bool use_divisor) { in avg_pool2d_out_cuda_frame()
88 const bool count_include_pad, const bool use_divisor) { in avg_pool2d_out_cuda_frame_nhwc()
137 bool count_include_pad, bool use_divisor) { in avg_pool2d_backward_out_cuda_frame()
194 bool count_include_pad, bool use_divisor) { in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| H A D | AveragePool3d.cu | 42 bool count_include_pad, in avg_pool3d_cuda_update_output()
111 bool count_include_pad, in avg_pool3d_cuda_update_output()
222 bool count_include_pad, in avg_pool3d_cuda_update_grad_input_atomic()
281 bool count_include_pad, int offsetZ, int divisor_override) in avg_pool3d_cuda_update_grad_input()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | pooling.h | 20 bool count_include_pad) { in avg_pool1d()
59 bool count_include_pad, in avg_pool2d()
106 bool count_include_pad, in avg_pool3d()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/ |
| H A D | functional.py | 49 count_include_pad=True, argument
95 count_include_pad=True, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AveragePool3d.cpp | 175 bool count_include_pad, in avg_pool3d_out_frame()
361 bool count_include_pad, in avg_pool3d_backward_out_frame()
|
| H A D | Pooling.cpp | 120 bool count_include_pad) { in avg_pool1d()
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ops/ |
| H A D | avgpool2d.py | 16 self, count_include_pad=False, ceil_mode=False, divisor_override=None argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_avg_pool2d.cpp | 30 bool count_include_pad, in avg_pool2d_out()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_caffe2.py | 266 count_include_pad, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 1672 …def test_avg_pool2d(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_overri… argument
1721 …def test_avg_pool2d_nhwc(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_o… argument
1783 …def test_avg_pool3d(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_overri… argument
1835 …def test_avg_pool3d_nhwc(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_o… argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 4541 count_include_pad=True, argument
4563 count_include_pad=True, argument
4584 count_include_pad, argument
4706 count_include_pad, argument
4875 count_include_pad, argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.cpp | 267 const bool count_include_pad, in check_avg_pool2d_args()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/serialization/ |
| H A D | mps_graph_schema.py | 108 count_include_pad: bool = True variable in MPSPooling2D
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2433 count_include_pad=True, argument
2556 count_include_pad, argument
2629 count_include_pad=True, argument
2716 count_include_pad, argument
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_pooling.py | 940 count_include_pad=True, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2015 bool count_include_pad,
2127 bool count_include_pad,
2175 bool count_include_pad,
|
| /aosp_15_r20/external/armnn/src/armnnOnnxParser/ |
| H A D | OnnxParser.cpp | 1587 uint32_t count_include_pad = 0; in ParseAveragePool() local
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 1595 def test_avgpool_2d(self, padding, count_include_pad): argument
|