Searched refs:im2col_buf (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 199 const int dims_out_depth, T* im2col_buf) { in operator ()()
201 MatrixMap C(im2col_buf, output_image_size, filter_total_size); in operator ()()
217 const int dims_out_depth, T* im2col_buf) { in operator ()()
251 im2col_buf, filter_total_size * output_image_size * sizeof(T)); in operator ()()
255 &ldA, out_data, &ldB, &beta, im2col_buf, &ldC); in operator ()()
509 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute() local
515 output_image_size, dims.out_depth, im2col_buf); in Compute()
517 Col2im<T>(im2col_buf, dims.in_depth, in Compute()
|
| D | conv_grad_ops_3d.cc | 532 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute() local
537 MatrixMap C(im2col_buf, output_image_size, filter_total_size); in Compute()
544 Col2im<T>(im2col_buf, dims.in_depth, in Compute()
|