| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | Pooling.cpp | 59 std::optional<int64_t> divisor_override) { in mkldnn_avg_pool2d()
69 std::optional<int64_t> divisor_override, in mkldnn_avg_pool2d_out()
81 std::optional<int64_t> divisor_override) { in mkldnn_avg_pool3d()
91 std::optional<int64_t> divisor_override, in mkldnn_avg_pool3d_out()
143 std::optional<int64_t> divisor_override, in mkldnn_avg_pool2d_backward_out()
156 std::optional<int64_t> divisor_override) { in mkldnn_avg_pool2d_backward()
167 std::optional<int64_t> divisor_override, in mkldnn_avg_pool3d_backward_out()
180 std::optional<int64_t> divisor_override) { in mkldnn_avg_pool3d_backward()
421 std::optional<int64_t> divisor_override) { in mkldnn_avg_pool2d()
446 std::optional<int64_t> divisor_override, in mkldnn_avg_pool2d_out()
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AvgPoolKernel.cpp | 24 std::optional<int64_t> divisor_override) { in cpu_avg_pool2d()
111 std::optional<int64_t> divisor_override) { in cpu_avg_pool2d_channels_last()
225 std::optional<int64_t> divisor_override) { in cpu_avg_pool2d_channels_last()
356 std::optional<int64_t> divisor_override) { in cpu_avg_pool2d_backward()
424 std::optional<int64_t> divisor_override) { in cpu_avg_pool2d_backward_channels_last()
503 std::optional<int64_t> divisor_override) { in avg_pool2d_kernel_impl()
529 std::optional<int64_t> divisor_override) { in avg_pool2d_backward_kernel_impl()
557 std::optional<int64_t> divisor_override) { in cpu_avg_pool3d()
653 std::optional<int64_t> divisor_override) { in cpu_avg_pool3d_channels_last()
776 std::optional<int64_t> divisor_override) { in cpu_avg_pool3d_channels_last()
[all …]
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_avg_pool2d_test.cpp | 29 exec_aten::optional<int64_t> divisor_override, in op_avg_pool2d_out()
108 exec_aten::optional<int64_t> divisor_override; in TEST_F() local
260 exec_aten::optional<int64_t> divisor_override = in TEST_F() local
491 exec_aten::optional<int64_t> divisor_override; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | AveragePool2d.cpp | 50 std::optional<int64_t> divisor_override) { in avg_pool2d_out_frame()
188 std::optional<int64_t> divisor_override) { in q_avg_pool2d()
268 std::optional<int64_t> divisor_override) { in qnnpack_avg_pool2d()
365 std::optional<int64_t> divisor_override) { in avg_pool2d_quantized_cpu()
|
| H A D | AveragePool3d.cpp | 103 std::optional<int64_t> divisor_override) { in q_avg_pool3d()
168 std::optional<int64_t> divisor_override) { in avg_pool3d_quantized_cpu()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Pool.cpp | 126 int32_t divisor_override; member
132 const ValueRef divisor_override, in create_divisor_params()
149 const ValueRef divisor_override, in add_avg_pool2d_node()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | AveragePool2d.cu | 38 scalar_t* const top_data, const int divisor_override, in avg_pool2d_out_cuda_frame()
87 scalar_t* const top_data, const int divisor_override, in avg_pool2d_out_cuda_frame_nhwc()
136 scalar_t* const bottom_diff, const int divisor_override, in avg_pool2d_backward_out_cuda_frame()
193 scalar_t* const bottom_diff, const int divisor_override, in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| H A D | AveragePool3d.cu | 43 int offsetZ, int divisor_override) in avg_pool3d_cuda_update_output()
112 int offsetZ, int divisor_override) in avg_pool3d_cuda_update_output()
223 int offsetZ, int divisor_override, const int gradInput_numel) 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/ao/nn/quantized/ |
| H A D | functional.py | 50 divisor_override=None, argument
96 divisor_override=None, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AveragePool3d.cpp | 176 std::optional<int64_t> divisor_override) in avg_pool3d_out_frame()
362 std::optional<int64_t> divisor_override) in avg_pool3d_backward_out_frame()
|
| /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 | 31 exec_aten::optional<int64_t> divisor_override, in avg_pool2d_out()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| H A D | pooling.h | 60 std::optional<int64_t> divisor_override) { in avg_pool2d()
107 std::optional<int64_t> divisor_override) { in avg_pool3d()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_caffe2.py | 267 divisor_override=None, argument
|
| H A D | symbolic_opset10.py | 409 divisor_override=None, argument
|
| H A D | symbolic_helper.py | 1427 divisor_override, argument
|
| H A D | symbolic_opset9.py | 1549 divisor_override=None, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/ |
| H A D | graph_helper.cpp | 253 auto divisor_override = toIValue(node->namedInput("divisor_override")); in createOperator() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 1672 … test_avg_pool2d(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_override): argument
1721 …_avg_pool2d_nhwc(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_override): argument
1783 … test_avg_pool3d(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_override): argument
1835 …_avg_pool3d_nhwc(self, X, kernel, stride, padding, ceil_mode, count_include_pad, divisor_override): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 4542 divisor_override=None, argument
4564 divisor_override=None, argument
4585 divisor_override, argument
4707 divisor_override=None, argument
4876 divisor_override=None, argument
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/util/ |
| H A D | kernel_ops_util.cpp | 268 const exec_aten::optional<int64_t>& divisor_override, in check_avg_pool2d_args()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/serialization/ |
| H A D | mps_graph_schema.py | 109 divisor_override: int = 0 variable in MPSPooling2D
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2434 divisor_override=None, argument
2557 divisor_override, argument
2630 divisor_override=None, argument
2717 divisor_override, argument
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_pooling.py | 941 divisor_override=None, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 2016 std::optional<int64_t> divisor_override) {
2128 std::optional<int64_t> divisor_override) {
2176 std::optional<int64_t> divisor_override) {
|