Searched refs:out_backprop_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 461 const T* out_backprop_data = out_backprop.template flat<T>().data(); in Compute() local
485 ConstTensorMap A(out_backprop_data + output_offset * image_id, in Compute()
529 &out_backprop_data, &filter_data, &input_offset, in Compute()
534 const T* out_data = out_backprop_data + shard_id * output_offset; in Compute()
567 out_backprop_data += output_offset * shard_limit; in Compute()
922 const T* out_backprop_data = out_backprop.template flat<T>().data(); in Compute() local
981 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
988 out_backprop_data += output_offset * shard_limit; in Compute()
|
| D | conv_grad_input_ops.cc | 457 const T* out_backprop_data = out_backprop.template flat<T>().data(); in Compute() local
481 ConstTensorMap A(out_backprop_data + output_offset * image_id, in Compute()
506 &out_backprop_data, &filter_data, &input_offset, in Compute()
511 const T* out_data = out_backprop_data + shard_id * output_offset; in Compute()
530 out_backprop_data += output_offset * shard_limit; in Compute()
|
| D | conv_grad_filter_ops.cc | 343 const T* out_backprop_data = out_backprop.template flat<T>().data(); in Compute() local
390 ConstTensorMap B(out_backprop_data, output_image_size * shard_limit, in Compute()
397 out_backprop_data += output_offset * shard_limit; in Compute()
|