/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_pooling_gpu.cc | 218 auto output_backprop_data = in Compute() local 232 output_backprop_data, &input_backprop_data) in Compute()
|
D | pooling_ops_common.cc | 406 auto output_backprop_data = in Compute() local 420 output_backprop_data, &input_backprop_data) in Compute()
|
D | cudnn_rnn_ops.cc | 829 auto output_backprop_data = AsDeviceMemory<T>(output_backprop); in DoBackward() local 854 output_backprop_data, output_h_backprop_data, in DoBackward()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 141 const DeviceMemory<Eigen::half>& output_backprop_data, 166 const DeviceMemory<float>& output_backprop_data, 191 const DeviceMemory<double>& output_backprop_data, 639 const DeviceMemory<T>& output_backprop_data,
|
D | cuda_dnn.cc | 1673 const DeviceMemory<T>& output_backprop_data, in DoRnnBackwardImpl() argument 1715 /*dy=*/output_backprop_data.opaque(), nullptr, nullptr, in DoRnnBackwardImpl() 1744 /*dy=*/output_backprop_data.opaque(), in DoRnnBackwardImpl() 2011 const DeviceMemory<Eigen::half>& output_backprop_data, in DoRnnBackward() argument 2041 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in DoRnnBackward() 2062 const DeviceMemory<float>& output_backprop_data, in DoRnnBackward() argument 2092 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in DoRnnBackward() 2114 const DeviceMemory<double>& output_backprop_data, in DoRnnBackward() argument 2144 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in DoRnnBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 132 const DeviceMemory<Eigen::half>& output_backprop_data, 157 const DeviceMemory<float>& output_backprop_data, 182 const DeviceMemory<double>& output_backprop_data, 694 const DeviceMemory<T>& output_backprop_data,
|
D | rocm_dnn.cc | 2130 const DeviceMemory<T>& output_backprop_data, in DoRnnBackwardImpl() argument 2189 output_backprop_data.opaque() /*dy*/, output_h_desc.handle() /*dhyDesc*/, in DoRnnBackwardImpl() 2446 const DeviceMemory<Eigen::half>& output_backprop_data, in DoRnnBackward() argument 2477 output_h_data, miopen_output_c_desc, output_c_data, output_backprop_data, in DoRnnBackward() 2497 const DeviceMemory<float>& output_backprop_data, in DoRnnBackward() argument 2528 output_h_data, miopen_output_c_desc, output_c_data, output_backprop_data, in DoRnnBackward() 2549 const DeviceMemory<double>& output_backprop_data, in DoRnnBackward() argument
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2234 const DeviceMemory<Eigen::half>& output_backprop_data, 2262 const DeviceMemory<float>& output_backprop_data, 2290 const DeviceMemory<double>& output_backprop_data,
|
D | stream.h | 1809 const DeviceMemory<Eigen::half> &output_backprop_data, 1834 const DeviceMemory<float> &output_backprop_data, 1859 const DeviceMemory<double> &output_backprop_data,
|
D | stream.cc | 4970 const DeviceMemory<Eigen::half> &output_backprop_data, in ThenRnnBackward() argument 4987 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward() 5016 const DeviceMemory<float> &output_backprop_data, in ThenRnnBackward() argument 5033 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward() 5063 const DeviceMemory<double> &output_backprop_data, in ThenRnnBackward() argument 5080 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward()
|