| /external/tensorflow/tensorflow/core/kernels/ |
| conv_grad_input_ops.cc | 456 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
479 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
488 col_buffer_data, dims.in_depth, dims.spatial_dims[0].input_size,
505 &input_backprop_data, &col_buffer_data,
509 T* im2col_buf = col_buffer_data + shard_id * size_C;
[all...] |
| conv_grad_ops_3d.cc | 460 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
483 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
491 Col2im<T>(col_buffer_data, dims.in_depth,
528 &input_backprop_data, &col_buffer_data,
532 T* im2col_buf = col_buffer_data + shard_id * size_C;
921 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
[all...] |
| conv_grad_filter_ops.cc | 342 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
368 auto shard = [&input_data, &col_buffer_data, &dims, &pad_top, &pad_left,
373 T* col_data_shard = col_buffer_data + shard_id * size_A;
388 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit,
[all...] |
