Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h88 const DeviceMemory<Eigen::half>& input_c_data,
107 const DeviceMemory<float>& input_c_data,
126 const DeviceMemory<double>& input_c_data,
145 const DeviceMemory<Eigen::half>& input_c_data,
171 const DeviceMemory<float>& input_c_data,
197 const DeviceMemory<double>& input_c_data,
575 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params,
595 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params,
Dcuda_dnn.cc1779 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument
1937 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument
1951 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2010 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
2068 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
2085 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
2098 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
2117 /*cx=*/input_c_data.opaque(), /*wDesc=*/rnn_desc.params_handle(), in DoRnnForwardImpl()
2150 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument
2170 input_h_data, input_c_desc, input_c_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h108 const DeviceMemory<Eigen::half>& input_c_data,
127 const DeviceMemory<float>& input_c_data,
146 const DeviceMemory<double>& input_c_data,
165 const DeviceMemory<Eigen::half>& input_c_data,
191 const DeviceMemory<float>& input_c_data,
217 const DeviceMemory<double>& input_c_data,
672 const DeviceMemory<T>& input_c_data,
689 const DeviceMemory<T>& input_c_data,
Drocm_dnn.cc2033 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward() argument
2143 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl() argument
2156 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2215 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2233 input_c_data.opaque() /*cx*/, rnn_desc.params_handle() /*wDesc*/, in DoRnnForwardImpl()
2258 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl() argument
2278 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
2331 input_c_desc.handle() /*cxDesc*/, input_c_data.opaque() /*cx*/, in DoRnnBackwardImpl()
2639 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward() argument
2669 miopen_input_h_desc, input_h_data, miopen_input_c_desc, input_c_data, in DoRnnForward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.cc2038 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward() argument
2053 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2074 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnForward() argument
2088 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2109 const DeviceMemory<double> &input_c_data, in ThenRnnForward() argument
2124 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnForward()
2145 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward() argument
2167 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnBackward()
2191 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnBackward() argument
2212 input_h_data, input_c_desc, input_c_data, params, output_desc, in ThenRnnBackward()
[all …]
Ddnn.h2120 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward() argument
2142 const DeviceMemory<float>& input_c_data, in DoRnnForward() argument
2164 const DeviceMemory<double>& input_c_data, in DoRnnForward() argument
2227 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward() argument
2256 const DeviceMemory<float>& input_c_data, in DoRnnBackward() argument
2285 const DeviceMemory<double>& input_c_data, in DoRnnBackward() argument
Dstream.h1233 const DeviceMemory<Eigen::half> &input_c_data,
1253 const DeviceMemory<float> &input_c_data,
1272 const DeviceMemory<double> &input_c_data,
1294 const DeviceMemory<Eigen::half> &input_c_data,
1320 const DeviceMemory<float> &input_c_data,
1346 const DeviceMemory<double> &input_c_data,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc818 DeviceMemory<T> input_c_data; in DoForward() local
820 input_c_data = AsDeviceMemory<T>(input_c); in DoForward()
855 input_c_data, params_data, *output_desc, in DoForward()
901 DeviceMemory<T> input_c_data; in DoBackward() local
903 input_c_data = AsDeviceMemory<T>(input_c); in DoBackward()
954 input_h_data, *c_state_desc, input_c_data, params_data, in DoBackward()