Searched defs:pad_d (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unfold3d.cpp | 238 int64_t pad_d, in Unfold3dCopyKernelImpl()
370 int64_t pad_d, in Unfold3dAccKernelImpl()
451 int64_t pad_d, in Unfold3dCopyCPU()
499 int64_t pad_d, in Unfold3dAccCPU()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PaddingKernel.cpp | 152 int64_t pad_d = ndim == 3 ? p.pads[ndim - 3] : 0; in cpu_padding() local
259 int64_t pad_d = ndim == 3 ? p.pads[ndim - 3] : 0; in cpu_padding_channels_last() local
333 int64_t pad_d = ndim == 3 ? p.pads[ndim - 3] : 0; in cpu_padding_backward() local
421 int64_t pad_d = ndim == 3 ? p.pads[ndim - 3] : 0; in cpu_padding_backward_channels_last() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_functional.py | 198 stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, dilation, X_scale, argument
|
| H A D | test_quantized_op.py | 6409 pad_d, argument
6493 pad_d, argument
6560 self, inputs, stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, o_pad, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv.cpp | 299 const int pad_d = kSpatialDim == 2 ? 0 : padding_[0]; in apply_impl() local
|