Searched refs:Col2im (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 65 void Col2im(const T* col_data, const int depth, const int height, in Col2im() function
487 Col2im<T>( in Compute()
517 Col2im<T>(im2col_buf, dims.in_depth, in Compute()
|
| D | conv_grad_ops_3d.cc | 65 void Col2im(const T* col_data, const int depth, const int planes, in Col2im() function
491 Col2im<T>(col_buffer_data, dims.in_depth, in Compute()
544 Col2im<T>(im2col_buf, dims.in_depth, in Compute()
|