Home
last modified time | relevance | path

Searched refs:input_h_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h77 const DeviceMemory<Eigen::half>& input_h_data,
95 const DeviceMemory<float>& input_h_data,
113 const DeviceMemory<double>& input_h_data,
131 const DeviceMemory<Eigen::half>& input_h_data,
156 const DeviceMemory<float>& input_h_data,
181 const DeviceMemory<double>& input_h_data,
611 const DeviceMemory<T>& input_h_data,
630 const DeviceMemory<T>& input_h_data,
Dcuda_dnn.cc1416 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward() argument
1511 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl() argument
1526 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
1572 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnForwardImpl()
1594 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
1609 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnForwardImpl()
1633 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
1664 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl() argument
1686 input_h_data, input_c_desc, input_c_data, in DoRnnBackwardImpl()
1721 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h68 const DeviceMemory<Eigen::half>& input_h_data,
86 const DeviceMemory<float>& input_h_data,
104 const DeviceMemory<double>& input_h_data,
122 const DeviceMemory<Eigen::half>& input_h_data,
147 const DeviceMemory<float>& input_h_data,
172 const DeviceMemory<double>& input_h_data,
667 const DeviceMemory<T>& input_h_data,
684 const DeviceMemory<T>& input_h_data,
Drocm_dnn.cc1896 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward() argument
2006 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl() argument
2020 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
2079 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2097 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2121 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl() argument
2142 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnBackwardImpl()
2194 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/, in DoRnnBackwardImpl()
2217 input_h_data.opaque() /*hx*/, output_desc.handles() /*yDesc*/, in DoRnnBackwardImpl()
2329 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2120 const DeviceMemory<Eigen::half>& input_h_data,
2141 const DeviceMemory<float>& input_h_data,
2162 const DeviceMemory<double>& input_h_data,
2224 const DeviceMemory<Eigen::half>& input_h_data,
2252 const DeviceMemory<float>& input_h_data,
2280 const DeviceMemory<double>& input_h_data,
Dstream.h1741 const DeviceMemory<Eigen::half> &input_h_data,
1760 const DeviceMemory<float> &input_h_data,
1778 const DeviceMemory<double> &input_h_data,
1799 const DeviceMemory<Eigen::half> &input_h_data,
1824 const DeviceMemory<float> &input_h_data,
1849 const DeviceMemory<double> &input_h_data,
Dstream.cc4850 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward() argument
4867 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4887 const DeviceMemory<float> &input_h_data, in ThenRnnForward() argument
4903 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4923 const DeviceMemory<double> &input_h_data, in ThenRnnForward() argument
4940 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4960 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward() argument
4984 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
5007 const DeviceMemory<float> &input_h_data, in ThenRnnBackward() argument
5030 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc755 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
773 input_h_data, *state_desc, input_c_data, params_data, in DoForward()
817 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local
851 input_h_data, *state_desc, input_c_data, in DoBackward()