Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h86 DeviceMemory<Eigen::half>* output_c_data, bool is_training,
104 DeviceMemory<float>* output_c_data, bool is_training,
122 DeviceMemory<double>* output_c_data, bool is_training,
140 const DeviceMemory<Eigen::half>& output_c_data,
165 const DeviceMemory<float>& output_c_data,
190 const DeviceMemory<double>& output_c_data,
619 DeviceMemory<T>* output_c_data, bool is_training,
638 const DeviceMemory<T>& output_c_data,
Dcuda_dnn.cc1424 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward() argument
1519 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument
1528 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1578 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1599 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl()
1615 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1638 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl()
1672 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument
1688 output_h_data, output_c_desc, output_c_data)); in DoRnnBackwardImpl()
1885 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h77 DeviceMemory<Eigen::half>* output_c_data, bool is_training,
95 DeviceMemory<float>* output_c_data, bool is_training,
113 DeviceMemory<double>* output_c_data, bool is_training,
131 const DeviceMemory<Eigen::half>& output_c_data,
156 const DeviceMemory<float>& output_c_data,
181 const DeviceMemory<double>& output_c_data,
676 DeviceMemory<T>* output_c_data, bool is_training,
693 const DeviceMemory<T>& output_c_data,
Drocm_dnn.cc1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() argument
2014 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument
2022 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2084 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl()
2102 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl()
2129 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument
2144 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
2338 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument
2363 output_h_data, miopen_output_c_desc, output_c_data, is_training, in DoRnnForward()
2380 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2129 DeviceMemory<Eigen::half>* output_c_data,
2150 DeviceMemory<float>* output_c_data,
2171 DeviceMemory<double>* output_c_data,
2233 const DeviceMemory<Eigen::half>& output_c_data,
2261 const DeviceMemory<float>& output_c_data,
2289 const DeviceMemory<double>& output_c_data,
Dstream.h1750 DeviceMemory<Eigen::half> *output_c_data,
1769 DeviceMemory<float> *output_c_data, bool is_training,
1787 DeviceMemory<double> *output_c_data, bool is_training,
1808 const DeviceMemory<Eigen::half> &output_c_data,
1833 const DeviceMemory<float> &output_c_data,
1858 const DeviceMemory<double> &output_c_data,
Dstream.cc4859 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward() argument
4869 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4895 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward() argument
4905 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4932 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward() argument
4942 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4969 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward() argument
4986 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
5015 const DeviceMemory<float> &output_c_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/
Dcudnn_rnn_ops.cc764 DeviceMemory<T> output_c_data; in DoForward() local
766 output_c_data = AsDeviceMemory<T>(output_c); in DoForward()
775 &output_h_data, *state_desc, &output_c_data, in DoForward()
825 DeviceMemory<T> output_c_data; in DoBackward() local
827 output_c_data = AsDeviceMemory<T>(output_c); in DoBackward()
853 output_h_data, *state_desc, output_c_data, in DoBackward()