Home
last modified time | relevance | path

Searched refs:output_c_backprop_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h148 const DeviceMemory<Eigen::half>& output_c_backprop_data,
174 const DeviceMemory<float>& output_c_backprop_data,
200 const DeviceMemory<double>& output_c_backprop_data,
633 const DeviceMemory<T>& output_c_backprop_data,
Dcuda_dnn.cc2099 const DeviceMemory<T>& output_c_backprop_data, in DoRnnBackwardImpl() argument
2160 /*dcy=*/output_c_backprop_data.opaque(), in DoRnnBackwardImpl()
2228 /*dcy=*/output_c_backprop_data.opaque(), in DoRnnBackwardImpl()
2252 /*dcy=*/output_c_backprop_data.opaque(), in DoRnnBackwardImpl()
2558 const DeviceMemory<Eigen::half>& output_c_backprop_data, in DoRnnBackward() argument
2587 output_c_backprop_data, input_backprop_data, input_h_backprop_data, in DoRnnBackward()
2610 const DeviceMemory<float>& output_c_backprop_data, in DoRnnBackward() argument
2639 output_c_backprop_data, input_backprop_data, input_h_backprop_data, in DoRnnBackward()
2663 const DeviceMemory<double>& output_c_backprop_data, in DoRnnBackward() argument
2692 output_c_backprop_data, input_backprop_data, input_h_backprop_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h175 const DeviceMemory<Eigen::half>& output_c_backprop_data,
201 const DeviceMemory<float>& output_c_backprop_data,
227 const DeviceMemory<double>& output_c_backprop_data,
709 const DeviceMemory<T>& output_c_backprop_data,
Drocm_dnn.cc2264 const DeviceMemory<T>& output_c_backprop_data, in DoRnnBackwardImpl() argument
2325 output_c_backprop_data.opaque() /*dcy*/, in DoRnnBackwardImpl()
2756 const DeviceMemory<Eigen::half>& output_c_backprop_data, in DoRnnBackward() argument
2786 output_h_backprop_data, output_c_backprop_data, input_backprop_data, in DoRnnBackward()
2808 const DeviceMemory<float>& output_c_backprop_data, in DoRnnBackward() argument
2838 output_h_backprop_data, output_c_backprop_data, input_backprop_data, in DoRnnBackward()
2861 const DeviceMemory<double>& output_c_backprop_data, in DoRnnBackward() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2211 const DeviceMemory<Eigen::half>& output_c_backprop_data, in DoRnnBackward() argument
2240 const DeviceMemory<float>& output_c_backprop_data, in DoRnnBackward() argument
2269 const DeviceMemory<double>& output_c_backprop_data, in DoRnnBackward() argument
Dstream.h1977 const DeviceMemory<Eigen::half> &output_c_backprop_data,
2003 const DeviceMemory<float> &output_c_backprop_data,
2029 const DeviceMemory<double> &output_c_backprop_data,
Dstream.cc4106 const DeviceMemory<Eigen::half> &output_c_backprop_data, in ThenRnnBackward() argument
4120 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward()
4151 const DeviceMemory<float> &output_c_backprop_data, in ThenRnnBackward() argument
4165 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward()
4197 const DeviceMemory<double> &output_c_backprop_data, in ThenRnnBackward() argument
4211 output_backprop_data, output_h_backprop_data, output_c_backprop_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc913 DeviceMemory<T> output_c_backprop_data; in DoBackward() local
915 output_c_backprop_data = AsDeviceMemory<T>(output_c_backprop); in DoBackward()
956 output_h_backprop_data, output_c_backprop_data, in DoBackward()