Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h86 const DeviceMemory<Eigen::half>& input_h_data,
105 const DeviceMemory<float>& input_h_data,
124 const DeviceMemory<double>& input_h_data,
143 const DeviceMemory<Eigen::half>& input_h_data,
169 const DeviceMemory<float>& input_h_data,
195 const DeviceMemory<double>& input_h_data,
573 const DeviceMemory<T>& input_h_data,
593 const DeviceMemory<T>& input_h_data,
Dcuda_dnn.cc1777 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward() argument
1935 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl() argument
1950 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
2008 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnForwardImpl()
2067 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnForwardImpl()
2084 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
2097 /*hxDesc=*/input_h_desc.handle(), /*hx=*/input_h_data.opaque(), in DoRnnForwardImpl()
2116 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
2148 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl() argument
2170 input_h_data, input_c_desc, input_c_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h106 const DeviceMemory<Eigen::half>& input_h_data,
125 const DeviceMemory<float>& input_h_data,
144 const DeviceMemory<double>& input_h_data,
163 const DeviceMemory<Eigen::half>& input_h_data,
189 const DeviceMemory<float>& input_h_data,
215 const DeviceMemory<double>& input_h_data,
670 const DeviceMemory<T>& input_h_data,
687 const DeviceMemory<T>& input_h_data,
Drocm_dnn.cc2031 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward() argument
2141 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl() argument
2155 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnForwardImpl()
2214 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2232 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2256 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl() argument
2277 rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in DoRnnBackwardImpl()
2330 input_h_desc.handle() /*hxDesc*/, input_h_data.opaque() /*hx*/, in DoRnnBackwardImpl()
2353 input_h_data.opaque() /*hx*/, output_desc.handles() /*yDesc*/, in DoRnnBackwardImpl()
2637 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.cc2036 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward() argument
2053 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2072 const DeviceMemory<float> &input_h_data, in ThenRnnForward() argument
2088 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2107 const DeviceMemory<double> &input_h_data, in ThenRnnForward() argument
2124 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2143 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward() argument
2167 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnBackward()
2189 const DeviceMemory<float> &input_h_data, in ThenRnnBackward() argument
2212 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnBackward()
[all …]
Ddnn.h2118 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward() argument
2140 const DeviceMemory<float>& input_h_data, in DoRnnForward() argument
2162 const DeviceMemory<double>& input_h_data, in DoRnnForward() argument
2225 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward() argument
2254 const DeviceMemory<float>& input_h_data, in DoRnnBackward() argument
2283 const DeviceMemory<double>& input_h_data, in DoRnnBackward() argument
Dstream.h1231 const DeviceMemory<Eigen::half> &input_h_data,
1251 const DeviceMemory<float> &input_h_data,
1270 const DeviceMemory<double> &input_h_data,
1292 const DeviceMemory<Eigen::half> &input_h_data,
1318 const DeviceMemory<float> &input_h_data,
1344 const DeviceMemory<double> &input_h_data,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc817 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
854 *h_state_desc, input_h_data, *c_state_desc, in DoForward()
900 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local
954 input_h_data, *c_state_desc, input_c_data, params_data, in DoBackward()