Searched refs:in_backprop_ptr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops_gpu.cu.cc | 84 int rate_cols, int pad_top, int pad_left, T* in_backprop_ptr) { in DilationBackpropInputKernel() argument
121 in_backprop_ptr + d + in DilationBackpropInputKernel()
|
| D | conv_grad_input_ops.cc | 922 auto in_backprop_ptr = in operator ()() local
969 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
998 input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
|
| D | depthwise_conv_grad_op.cc | 643 auto in_backprop_ptr = in_backprop->template flat<T>().data(); in Compute() local
645 context, args, out_backprop_ptr, filter_ptr, in_backprop_ptr, in Compute()
|
| D | conv_grad_ops_3d.cc | 1331 auto in_backprop_ptr = in Compute() local
1377 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1413 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
|