Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2094 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2116 const DeviceMemory<float>& input_c_data, in DoRnnForward()
2138 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2201 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2230 const DeviceMemory<float>& input_c_data, in DoRnnBackward()
2259 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
Dstream.cc3989 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
4025 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnForward()
4060 const DeviceMemory<double> &input_c_data, in ThenRnnForward()
4096 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward()
4142 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnBackward()
4187 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2030 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
2140 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2255 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2636 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2680 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2723 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2746 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2799 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2851 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1721 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
1877 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2090 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2417 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2461 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2504 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2548 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2601 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2653 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc817 DeviceMemory<T> input_c_data; in DoForward() local
900 DeviceMemory<T> input_c_data; in DoBackward() local