| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 84 DeviceMemory<Eigen::half>* output_h_data,
102 DeviceMemory<float>* output_h_data,
120 DeviceMemory<double>* output_h_data,
138 const DeviceMemory<Eigen::half>& output_h_data,
163 const DeviceMemory<float>& output_h_data,
188 const DeviceMemory<double>& output_h_data,
617 DeviceMemory<T>* output_h_data,
636 const DeviceMemory<T>& output_h_data,
|
| D | cuda_dnn.cc | 1422 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward() argument
1517 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl() argument
1528 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1577 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1598 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1614 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1637 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1670 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl() argument
1688 output_h_data, output_c_desc, output_c_data)); in DoRnnBackwardImpl()
1883 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward() argument
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 75 DeviceMemory<Eigen::half>* output_h_data,
93 DeviceMemory<float>* output_h_data,
111 DeviceMemory<double>* output_h_data,
129 const DeviceMemory<Eigen::half>& output_h_data,
154 const DeviceMemory<float>& output_h_data,
179 const DeviceMemory<double>& output_h_data,
674 DeviceMemory<T>* output_h_data,
691 const DeviceMemory<T>& output_h_data,
|
| D | rocm_dnn.cc | 1902 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward() argument
2012 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl() argument
2022 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2083 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2101 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2127 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl() argument
2144 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
2336 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward() argument
2363 output_h_data, miopen_output_c_desc, output_c_data, is_training, in DoRnnForward()
2378 DeviceMemory<float>* output_h_data, in DoRnnForward() argument
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2127 DeviceMemory<Eigen::half>* output_h_data,
2148 DeviceMemory<float>* output_h_data,
2169 DeviceMemory<double>* output_h_data,
2231 const DeviceMemory<Eigen::half>& output_h_data,
2259 const DeviceMemory<float>& output_h_data,
2287 const DeviceMemory<double>& output_h_data,
|
| D | stream.h | 1748 DeviceMemory<Eigen::half> *output_h_data,
1767 DeviceMemory<float> *output_h_data,
1785 DeviceMemory<double> *output_h_data,
1806 const DeviceMemory<Eigen::half> &output_h_data,
1831 const DeviceMemory<float> &output_h_data,
1856 const DeviceMemory<double> &output_h_data,
|
| D | stream.cc | 4857 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward() argument
4869 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4893 DeviceMemory<float> *output_h_data, in ThenRnnForward() argument
4905 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4930 DeviceMemory<double> *output_h_data, in ThenRnnForward() argument
4942 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4967 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward() argument
4986 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
5013 const DeviceMemory<float> &output_h_data, in ThenRnnBackward() argument
5032 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 | 763 auto output_h_data = AsDeviceMemory<T>(output_h); in DoForward() local
775 &output_h_data, *state_desc, &output_c_data, in DoForward()
824 auto output_h_data = AsDeviceMemory<T>(output_h); in DoBackward() local
853 output_h_data, *state_desc, output_c_data, in DoBackward()
|