| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops_gpu.cu.cc | 79 const T* __restrict__ filter_ptr, const T* __restrict__ out_backprop_ptr, in DilationBackpropInputKernel() argument
122 out_backprop_ptr[out_idx]); in DilationBackpropInputKernel()
129 const T* __restrict__ filter_ptr, const T* __restrict__ out_backprop_ptr, in DilationBackpropFilterKernel() argument
171 out_backprop_ptr[out_idx]); in DilationBackpropFilterKernel()
|
| D | conv_grad_input_ops.cc | 346 auto out_backprop_ptr = in operator ()() local
421 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
449 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
474 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
494 in_backprop_ptr, filter_ptr, out_backprop_ptr, input_desc, filter_desc, in operator ()()
501 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_grad_ops_3d.cc | 1384 auto out_backprop_ptr = in Compute() local
1455 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1486 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1493 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1513 filter_ptr, out_backprop_ptr, input_desc, in Compute()
1524 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1883 auto out_backprop_ptr = in Compute() local
1939 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
1962 filter_desc, filter_backprop_ptr, output_desc, out_backprop_ptr, in Compute()
1970 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
[all …]
|
| D | conv_grad_filter_ops.cc | 943 se::DeviceMemory<T> out_backprop_ptr = in operator ()() local
1016 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1047 filter_desc, filter_backprop_ptr, output_desc, out_backprop_ptr, in operator ()()
1072 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
1092 filter_backprop_ptr, out_backprop_ptr, input_desc, in operator ()()
1101 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | avgpooling_op.cc | 331 const T* out_backprop_ptr = out_backprop.flat<T>().data(); in Compute() local
334 auto shard = [context, out_backprop_ptr, input_backprop_ptr, in Compute()
367 out_backprop_ptr + output_index * out_backprop_depth; in Compute()
|
| D | depthwise_conv_grad_op.cc | 682 auto out_backprop_ptr = out_backprop.template flat<T>().data(); in Compute() local
686 context, args, out_backprop_ptr, filter_ptr, in_backprop_ptr, in Compute()
1206 auto out_backprop_ptr = casted_out_backprop.template flat<U>().data(); in Compute() local
1210 context, args, out_backprop_ptr, input_ptr, filter_backprop_ptr, in Compute()
|