Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2101 DeviceMemory<Eigen::half>* output_c_data, in DoRnnForward()
2123 DeviceMemory<float>* output_c_data, in DoRnnForward()
2145 DeviceMemory<double>* output_c_data, in DoRnnForward()
2208 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2237 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2266 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
Dstream.cc3996 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward()
4031 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward()
4067 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward()
4103 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward()
4148 const DeviceMemory<float> &output_c_data, in ThenRnnBackward()
4194 const DeviceMemory<double> &output_c_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2036 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward()
2146 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
2261 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
2643 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
2686 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
2730 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2753 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2805 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2858 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1727 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward()
1883 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
2096 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
2424 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
2467 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
2511 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2555 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2607 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2660 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc825 DeviceMemory<T> output_c_data; in DoForward() local
907 DeviceMemory<T> output_c_data; in DoBackward() local