| /external/tensorflow/tensorflow/core/kernels/ |
| dilation_ops_gpu.cu.cc | 81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols,
123 out_backprop_ptr[out_idx]);
130 const T* out_backprop_ptr, int batch, int input_rows, int input_cols,
171 out_backprop_ptr[out_idx]);
|
| avgpooling_op.cc | 287 const T* out_backprop_ptr = out_backprop.flat<T>().data(); variable
290 auto shard = [context, out_backprop_ptr, input_backprop_ptr,
323 out_backprop_ptr + output_index * out_backprop_depth;
|
| conv_grad_ops_3d.cc | 627 auto out_backprop_ptr = variable
996 auto out_backprop_ptr = variable
[all...] |
| depthwise_conv_grad_op.cc | 574 auto out_backprop_ptr = out_backprop.template flat<T>().data(); variable
582 context, args, out_backprop_ptr, filter_ptr, in_backprop_ptr,
952 auto out_backprop_ptr = out_backprop.template flat<T>().data(); variable
[all...] |
| conv_grad_filter_ops.cc | 876 auto out_backprop_ptr = local
[all...] |
| conv_grad_input_ops.cc | 923 auto out_backprop_ptr = local
[all...] |
