Searched refs:col_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_filter_ops.cc | 68 const int stride_h, const int stride_w, T* col_data) { in Im2col() argument
79 memcpy(col_data, input_data + (ih * width + iw) * depth, in Im2col()
83 memset(col_data, 0, sizeof(T) * depth); in Im2col()
85 col_data += depth; in Im2col()
|
| D | conv_grad_ops_3d.cc | 65 void Col2im(const T* col_data, const int depth, const int planes, in Col2im() argument
88 im_patch_data[i] += col_data[i]; in Col2im()
92 col_data += depth; in Col2im()
119 const int stride_w, T* col_data) { in Im2col() argument
135 memcpy(col_data, in Im2col()
141 memset(col_data, 0, sizeof(T) * depth); in Im2col()
143 col_data += depth; in Im2col()
|
| D | conv_grad_input_ops.cc | 65 void Col2im(const T* col_data, const int depth, const int height, in Col2im() argument
81 im_patch_data[i] += col_data[i]; in Col2im()
85 col_data += depth; in Col2im()
|