Home
last modified time | relevance | path

Searched refs:output_backprop_data (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc218 auto output_backprop_data = in Compute() local
232 output_backprop_data, &input_backprop_data) in Compute()
Dpooling_ops_common.cc406 auto output_backprop_data = in Compute() local
420 output_backprop_data, &input_backprop_data) in Compute()
Dcudnn_rnn_ops.cc829 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/
Dcuda_dnn.h141 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,
Dcuda_dnn.cc1673 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/
Drocm_dnn.h132 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,
Drocm_dnn.cc2130 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/
Ddnn.h2234 const DeviceMemory<Eigen::half>& output_backprop_data,
2262 const DeviceMemory<float>& output_backprop_data,
2290 const DeviceMemory<double>& output_backprop_data,
Dstream.h1809 const DeviceMemory<Eigen::half> &output_backprop_data,
1834 const DeviceMemory<float> &output_backprop_data,
1859 const DeviceMemory<double> &output_backprop_data,
Dstream.cc4970 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()