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.h80 const DeviceMemory<Eigen::half>& input_c_data,
98 const DeviceMemory<float>& input_c_data,
116 const DeviceMemory<double>& input_c_data,
134 const DeviceMemory<Eigen::half>& input_c_data,
159 const DeviceMemory<float>& input_c_data,
184 const DeviceMemory<double>& input_c_data,
647 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params,
666 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params,
Dcuda_dnn.cc1604 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument
1753 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument
1767 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
1811 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1828 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
1842 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1861 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
1893 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument
1913 input_h_data, input_c_desc, input_c_data, in DoRnnBackwardImpl()
1948 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h107 const DeviceMemory<Eigen::half>& input_c_data,
125 const DeviceMemory<float>& input_c_data,
143 const DeviceMemory<double>& input_c_data,
161 const DeviceMemory<Eigen::half>& input_c_data,
186 const DeviceMemory<float>& input_c_data,
211 const DeviceMemory<double>& input_c_data,
744 const DeviceMemory<T>& input_c_data,
761 const DeviceMemory<T>& input_c_data,
Drocm_dnn.cc1979 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument
2089 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument
2102 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2161 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2179 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2204 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument
2224 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
2277 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/, in DoRnnBackwardImpl()
2584 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward() argument
2614 miopen_input_h_desc, input_h_data, miopen_input_c_desc, input_c_data, in DoRnnForward()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2191 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward() argument
2212 const DeviceMemory<float>& input_c_data, in DoRnnForward() argument
2233 const DeviceMemory<double>& input_c_data, in DoRnnForward() argument
2295 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward() argument
2323 const DeviceMemory<float>& input_c_data, in DoRnnBackward() argument
2351 const DeviceMemory<double>& input_c_data, in DoRnnBackward() argument
Dstream.h1785 const DeviceMemory<Eigen::half> &input_c_data,
1804 const DeviceMemory<float> &input_c_data,
1822 const DeviceMemory<double> &input_c_data,
1843 const DeviceMemory<Eigen::half> &input_c_data,
1868 const DeviceMemory<float> &input_c_data,
1893 const DeviceMemory<double> &input_c_data,
Dstream.cc4545 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward() argument
4560 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4579 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnForward() argument
4593 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4612 const DeviceMemory<double> &input_c_data, in ThenRnnForward() argument
4627 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4646 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward() argument
4668 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
4691 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnBackward() argument
4712 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc815 DeviceMemory<T> input_c_data; in DoForward() local
817 input_c_data = AsDeviceMemory<T>(input_c); in DoForward()
832 input_h_data, *c_state_desc, input_c_data, in DoForward()
878 DeviceMemory<T> input_c_data; in DoBackward() local
880 input_c_data = AsDeviceMemory<T>(input_c); in DoBackward()
912 *c_state_desc, input_c_data, params_data, *output_desc, in DoBackward()