/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 79 const DeviceMemory<Eigen::half>& input_c_data, 97 const DeviceMemory<float>& input_c_data, 115 const DeviceMemory<double>& input_c_data, 133 const DeviceMemory<Eigen::half>& input_c_data, 158 const DeviceMemory<float>& input_c_data, 183 const DeviceMemory<double>& input_c_data, 613 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, 632 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params,
|
D | cuda_dnn.cc | 1418 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument 1513 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument 1527 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl() 1573 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl() 1595 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl() 1610 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl() 1634 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl() 1666 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument 1686 input_h_data, input_c_desc, input_c_data, in DoRnnBackwardImpl() 1722 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnBackwardImpl() [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 70 const DeviceMemory<Eigen::half>& input_c_data, 88 const DeviceMemory<float>& input_c_data, 106 const DeviceMemory<double>& input_c_data, 124 const DeviceMemory<Eigen::half>& input_c_data, 149 const DeviceMemory<float>& input_c_data, 174 const DeviceMemory<double>& input_c_data, 669 const DeviceMemory<T>& input_c_data, 686 const DeviceMemory<T>& input_c_data,
|
D | rocm_dnn.cc | 1898 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument 2008 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument 2021 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl() 2080 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl() 2098 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl() 2123 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument 2143 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl() 2195 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/, in DoRnnBackwardImpl() 2331 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward() argument 2361 miopen_input_h_desc, input_h_data, miopen_input_c_desc, input_c_data, in DoRnnForward() [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2122 const DeviceMemory<Eigen::half>& input_c_data, 2143 const DeviceMemory<float>& input_c_data, 2164 const DeviceMemory<double>& input_c_data, 2226 const DeviceMemory<Eigen::half>& input_c_data, 2254 const DeviceMemory<float>& input_c_data, 2282 const DeviceMemory<double>& input_c_data,
|
D | stream.h | 1743 const DeviceMemory<Eigen::half> &input_c_data, 1762 const DeviceMemory<float> &input_c_data, 1780 const DeviceMemory<double> &input_c_data, 1801 const DeviceMemory<Eigen::half> &input_c_data, 1826 const DeviceMemory<float> &input_c_data, 1851 const DeviceMemory<double> &input_c_data,
|
D | stream.cc | 4852 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward() argument 4868 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward() 4889 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnForward() argument 4904 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward() 4925 const DeviceMemory<double> &input_c_data, in ThenRnnForward() argument 4941 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward() 4962 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward() argument 4985 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward() 5009 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnBackward() argument 5031 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 756 DeviceMemory<T> input_c_data; in DoForward() local 758 input_c_data = AsDeviceMemory<T>(input_c); in DoForward() 773 input_h_data, *state_desc, input_c_data, params_data, in DoForward() 818 DeviceMemory<T> input_c_data; in DoBackward() local 820 input_c_data = AsDeviceMemory<T>(input_c); in DoBackward() 851 input_h_data, *state_desc, input_c_data, in DoBackward()
|