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.h2191 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2212 const DeviceMemory<float>& input_c_data, in DoRnnForward()
2233 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2295 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2323 const DeviceMemory<float>& input_c_data, in DoRnnBackward()
2351 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
Dstream.cc4545 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
4579 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnForward()
4612 const DeviceMemory<double> &input_c_data, in ThenRnnForward()
4646 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward()
4691 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnBackward()
4735 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1604 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
1753 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
1893 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2133 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2175 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2216 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2258 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2310 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2361 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1979 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
2089 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2204 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2584 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2627 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2669 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2691 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2743 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2794 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc815 DeviceMemory<T> input_c_data; in DoForward() local
878 DeviceMemory<T> input_c_data; in DoBackward() local