Searched refs:out_backprop_ptr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops_gpu.cu.cc | 81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, in DilationBackpropInputKernel() argument
123 out_backprop_ptr[out_idx]); in DilationBackpropInputKernel()
130 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, in DilationBackpropFilterKernel() argument
171 out_backprop_ptr[out_idx]); in DilationBackpropFilterKernel()
|
| D | avgpooling_op.cc | 294 const T* out_backprop_ptr = out_backprop.flat<T>().data(); in Compute() local
297 auto shard = [context, out_backprop_ptr, input_backprop_ptr, in Compute()
330 out_backprop_ptr + output_index * out_backprop_depth; in Compute()
|
| D | depthwise_conv_grad_op.cc | 641 auto out_backprop_ptr = out_backprop.template flat<T>().data(); in Compute() local
645 context, args, out_backprop_ptr, filter_ptr, in_backprop_ptr, in Compute()
1124 auto out_backprop_ptr = out_backprop.template flat<T>().data(); in Compute() local
1128 context, args, out_backprop_ptr, input_ptr, filter_backprop_ptr, in Compute()
|
| D | conv_grad_ops_3d.cc | 1325 auto out_backprop_ptr = in Compute() local
1376 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1412 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1731 auto out_backprop_ptr = in Compute() local
1782 input_desc, input_ptr, output_desc, out_backprop_ptr, in Compute()
1819 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
|
| D | conv_grad_filter_ops.cc | 806 auto out_backprop_ptr = in operator ()() local
856 input_desc, input_ptr, output_desc, out_backprop_ptr, in operator ()()
887 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_grad_input_ops.cc | 916 auto out_backprop_ptr = in operator ()() local
968 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in operator ()()
997 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|