| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 85 DeviceMemory<Eigen::half>* output_h_data,
103 DeviceMemory<float>* output_h_data,
121 DeviceMemory<double>* output_h_data,
139 const DeviceMemory<Eigen::half>& output_h_data,
164 const DeviceMemory<float>& output_h_data,
189 const DeviceMemory<double>& output_h_data,
651 DeviceMemory<T>* output_h_data,
670 const DeviceMemory<T>& output_h_data,
|
| D | cuda_dnn.cc | 1608 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward() argument
1757 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl() argument
1768 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1815 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1831 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1846 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1864 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1897 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl() argument
1915 output_h_data, output_c_desc, output_c_data)); in DoRnnBackwardImpl()
2138 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward() argument
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 112 DeviceMemory<Eigen::half>* output_h_data,
130 DeviceMemory<float>* output_h_data,
148 DeviceMemory<double>* output_h_data,
166 const DeviceMemory<Eigen::half>& output_h_data,
191 const DeviceMemory<float>& output_h_data,
216 const DeviceMemory<double>& output_h_data,
749 DeviceMemory<T>* output_h_data,
766 const DeviceMemory<T>& output_h_data,
|
| D | rocm_dnn.cc | 1983 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward() argument
2093 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl() argument
2103 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2164 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2182 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2208 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl() argument
2225 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
2589 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward() argument
2616 output_h_data, miopen_output_c_desc, output_c_data, is_training, in DoRnnForward()
2631 DeviceMemory<float>* output_h_data, in DoRnnForward() argument
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2196 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward() argument
2217 DeviceMemory<float>* output_h_data, in DoRnnForward() argument
2238 DeviceMemory<double>* output_h_data, in DoRnnForward() argument
2300 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward() argument
2328 const DeviceMemory<float>& output_h_data, in DoRnnBackward() argument
2356 const DeviceMemory<double>& output_h_data, in DoRnnBackward() argument
|
| D | stream.h | 1790 DeviceMemory<Eigen::half> *output_h_data,
1809 DeviceMemory<float> *output_h_data,
1827 DeviceMemory<double> *output_h_data,
1848 const DeviceMemory<Eigen::half> &output_h_data,
1873 const DeviceMemory<float> &output_h_data,
1898 const DeviceMemory<double> &output_h_data,
|
| D | stream.cc | 4550 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward() argument
4561 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4583 DeviceMemory<float> *output_h_data, in ThenRnnForward() argument
4594 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4617 DeviceMemory<double> *output_h_data, in ThenRnnForward() argument
4628 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4651 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward() argument
4669 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
4695 const DeviceMemory<float> &output_h_data, in ThenRnnBackward() argument
4713 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 822 auto output_h_data = AsDeviceMemory<T>(output_h); in DoForward() local
834 *h_state_desc, &output_h_data, *c_state_desc, in DoForward()
884 auto output_h_data = AsDeviceMemory<T>(output_h); in DoBackward() local
913 output_data, *h_state_desc, output_h_data, *c_state_desc, in DoBackward()
|