Searched defs:data_col (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | im2col.cuh | 36 dt* data_col) { in im2col_kernel()
81 dt* data_col) { in im2col()
108 const dt* data_col, in col2im_device()
163 const dt* data_col, in col2im_kernel()
202 const dt* data_col, in col2im()
246 const dt* data_col, in col2im_batched_kernel()
294 const dt* data_col, in col2im_batched()
|
| H A D | vol2col.cuh | 39 T* data_col) { in vol2col_kernel()
97 T* data_col) { in vol2col()
131 const T* data_col, in vol2im_kernel()
201 const T* data_col, in col2vol()
|
| H A D | NaiveDilatedConvolution.cu | 41 Dtype* data_col) { in hvol2col()
92 const Dtype* data_col, in col2hvol()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | vol2col.h | 29 T* data_col) { in vol2col()
60 const T* data_col, in col2vol()
|
| H A D | NaiveDilatedConvolution.cpp | 37 Dtype* data_col, in hvol2col()
87 const Dtype* data_col, in col2hvol()
|