Searched refs:in_backprop_ptr (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 352 auto in_backprop_ptr = in operator ()() local
400 WrapRedzoneBestEffort(&rz_allocator, in_backprop_ptr)); in operator ()()
448 se::dnn::ToDataType<T>::value, stream, input_desc, in_backprop_ptr, in operator ()()
475 input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
494 in_backprop_ptr, filter_ptr, out_backprop_ptr, input_desc, filter_desc, in operator ()()
502 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in operator ()()
|
| D | dilation_ops_gpu.cu.cc | 83 T* __restrict__ in_backprop_ptr) { in DilationBackpropInputKernel() argument
120 in_backprop_ptr + d + in DilationBackpropInputKernel()
|
| D | conv_grad_ops_3d.cc | 1390 auto in_backprop_ptr = in Compute() local
1433 WrapRedzoneBestEffort(&rz_allocator, in_backprop_ptr)); in Compute()
1485 se::dnn::ToDataType<T>::value, stream, input_desc, in_backprop_ptr, in Compute()
1494 input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1512 se::dnn::ToDataType<T>::value, in_backprop_ptr, in Compute()
1525 input_desc, &in_backprop_ptr, &scratch_allocator, algorithm_config, in Compute()
|
| D | depthwise_conv_grad_op.cc | 684 auto in_backprop_ptr = in_backprop->template flat<T>().data(); in Compute() local
686 context, args, out_backprop_ptr, filter_ptr, in_backprop_ptr, in Compute()
|