Searched defs:kernel_d (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unfold3d.cpp | 188 int64_t kernel_d, in Unfold3dZeroPaddingCopyKernelImpl()
232 int64_t kernel_d, in Unfold3dCopyKernelImpl()
311 int64_t kernel_d, in Unfold3dZeroPaddingAccKernelImpl()
364 int64_t kernel_d, in Unfold3dAccKernelImpl()
445 int64_t kernel_d, in Unfold3dCopyCPU()
493 int64_t kernel_d, in Unfold3dAccCPU()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qconv_prepack.cpp | 71 const int kernel_d = kSpatialDim == 2 ? 1 : weight.size(2); in prepack() local
238 const uint32_t kernel_d = kSpatialDim == 3 ? weight.size(2) : 1; in prepack() local
|
| H A D | qconv_unpack_impl.cpp | 22 const int kernel_d = kSpatialDim == 2 ? 1 : kernel[0]; in unpack() local
|
| H A D | qconv.cpp | 296 const int kernel_d = kSpatialDim == 2 ? 1 : kernel[0]; in apply_impl() local
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_functional.py | 197 out_channels_per_group, groups, kernel_d, kernel_h, kernel_w, argument
|
| H A D | test_quantized_op.py | 6403 kernel_d, argument
6487 kernel_d, argument
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 918 input_d, kernel_d, dilation_d, padding_d, stride_d, device argument
|