Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h2127 DeviceMemory<Eigen::half>* output_c_data, in DoRnnForward()
2149 DeviceMemory<float>* output_c_data, in DoRnnForward()
2171 DeviceMemory<double>* output_c_data, in DoRnnForward()
2234 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2263 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2292 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
Dstream.cc2045 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward()
2080 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward()
2116 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward()
2152 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward()
2197 const DeviceMemory<float> &output_c_data, in ThenRnnBackward()
2243 const DeviceMemory<double> &output_c_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2039 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward()
2149 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
2264 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
2646 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
2689 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
2733 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2756 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2808 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2861 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1785 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward()
1943 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl()
2156 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl()
2484 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward()
2527 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward()
2571 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward()
2615 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward()
2667 const DeviceMemory<float>& output_c_data, in DoRnnBackward()
2720 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc826 DeviceMemory<T> output_c_data; in DoForward() local
908 DeviceMemory<T> output_c_data; in DoBackward() local