Searched refs:col_buffer_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 460 T* col_buffer_data = col_buffer.template flat<T>().data(); in Compute() local
483 TensorMap C(col_buffer_data, output_image_size, filter_total_size); in Compute()
491 Col2im<T>(col_buffer_data, dims.in_depth, in Compute()
528 &input_backprop_data, &col_buffer_data, in Compute()
532 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
921 T* col_buffer_data = col_buffer.template flat<T>().data(); in Compute() local
947 auto shard = [&input_data, &col_buffer_data, &dims, &top_pad_planes, in Compute()
953 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
979 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
|
| D | conv_grad_input_ops.cc | 456 T* col_buffer_data = col_buffer.template flat<T>().data(); in Compute() local
479 TensorMap C(col_buffer_data, output_image_size, filter_total_size); in Compute()
488 col_buffer_data, dims.in_depth, dims.spatial_dims[0].input_size, in Compute()
505 &input_backprop_data, &col_buffer_data, in Compute()
509 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
|
| D | conv_grad_filter_ops.cc | 342 T* col_buffer_data = col_buffer.template flat<T>().data(); in Compute() local
368 auto shard = [&input_data, &col_buffer_data, &dims, &pad_top, &pad_left, in Compute()
373 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
388 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit, in Compute()
|