Searched defs:width_col (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | im2col.cuh | 35 const int64_t width_col, in im2col_kernel()
72 const int64_t width_col, in im2col()
121 const int64_t width_col, in col2im_device()
176 const int64_t width_col, in col2im_kernel()
207 const int64_t width_col, in col2im()
261 const int64_t width_col, in col2im_batched_kernel()
301 const int64_t width_col, in col2im_batched()
|
| H A D | vol2col.cuh | 38 const int width_col, in vol2col_kernel()
84 const int width_col, in vol2col()
150 const unsigned width_col, in vol2im_kernel()
|
| H A D | Col2Im.cu | 111 int64_t width_col = (output_width + 2 * pad_width - in col2im_out_cuda_template() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | im2col.h | 33 const int64_t width_col = output_width; variable
107 const int64_t width_col = output_width; variable
|
| H A D | vol2col.h | 16 const int64_t width_col, in vol2col()
84 int64_t width_col = out_width; in col2vol() local
|
| H A D | Col2Im.cpp | 156 int64_t width_col = (output_width + 2 * pad_width - in col2im_out_cpu_template() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_grad_input_ops.h | 83 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im() local
|
| H A D | conv_grad_ops_3d.cc | 86 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im() local
135 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col() local
|
| H A D | conv_grad_filter_ops.cc | 79 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 4906 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im() local
8155 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im() local
|