Home
last modified time | relevance | path

Searched defs:output_c_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1424 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward()
1519 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
1672 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
1885 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
1926 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
1968 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2010 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2061 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2113 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward()
2014 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
2129 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
2338 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
2380 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
2423 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2445 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2496 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2548 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc764 DeviceMemory<T> output_c_data; in DoForward() local
825 DeviceMemory<T> output_c_data; in DoBackward() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4859 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward()
4895 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward()
4932 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward()
4969 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward()
5015 const DeviceMemory<float> &output_c_data, in ThenRnnBackward()
5062 const DeviceMemory<double> &output_c_data, in ThenRnnBackward()