Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h79 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,
Dcuda_dnn.cc1418 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/
Drocm_dnn.h70 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,
Drocm_dnn.cc1898 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/
Ddnn.h2122 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,
Dstream.h1743 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,
Dstream.cc4852 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> &params, 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> &params, in ThenRnnBackward() argument
5031 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc756 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()