/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 148 DeviceMemory<Eigen::half>* params_backprop_data, 173 DeviceMemory<float>* params_backprop_data, 198 DeviceMemory<double>* params_backprop_data, 679 DeviceMemory<T>* params_backprop_data,
|
D | cuda_dnn.cc | 1906 DeviceMemory<T>* params_backprop_data, in DoRnnBackwardImpl() argument 1984 if (params_backprop_data != nullptr) { in DoRnnBackwardImpl() 1986 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl() 1997 /*dw=*/params_backprop_data->opaque(), in DoRnnBackwardImpl() 2011 /*dw=*/params_backprop_data->opaque(), in DoRnnBackwardImpl() 2272 DeviceMemory<Eigen::half>* params_backprop_data, in DoRnnBackward() argument 2298 params_backprop_data, reserve_space_data, workspace_allocator, in DoRnnBackward() 2323 DeviceMemory<float>* params_backprop_data, in DoRnnBackward() argument 2349 params_backprop_data, reserve_space_data, workspace_allocator, in DoRnnBackward() 2375 DeviceMemory<double>* params_backprop_data, in DoRnnBackward() argument [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 175 DeviceMemory<Eigen::half>* params_backprop_data, 200 DeviceMemory<float>* params_backprop_data, 225 DeviceMemory<double>* params_backprop_data, 775 DeviceMemory<T>* params_backprop_data,
|
D | rocm_dnn.cc | 2217 DeviceMemory<T>* params_backprop_data, in DoRnnBackwardImpl() argument 2291 if (params_backprop_data != nullptr) { in DoRnnBackwardImpl() 2293 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl() 2301 params_backprop_data->opaque() /*dw*/, workspace.opaque() /*workspace*/, in DoRnnBackwardImpl() 2705 DeviceMemory<Eigen::half>* params_backprop_data, in DoRnnBackward() argument 2732 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward() 2756 DeviceMemory<float>* params_backprop_data, in DoRnnBackward() argument 2783 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward() 2808 DeviceMemory<double>* params_backprop_data, in DoRnnBackward() argument
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2309 DeviceMemory<Eigen::half>* params_backprop_data, in DoRnnBackward() argument 2337 DeviceMemory<float>* params_backprop_data, in DoRnnBackward() argument 2365 DeviceMemory<double>* params_backprop_data, in DoRnnBackward() argument
|
D | stream.h | 1857 DeviceMemory<Eigen::half> *params_backprop_data, 1882 DeviceMemory<float> *params_backprop_data, 1907 DeviceMemory<double> *params_backprop_data,
|
D | stream.cc | 4660 DeviceMemory<Eigen::half> *params_backprop_data, in ThenRnnBackward() argument 4672 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward() 4704 DeviceMemory<float> *params_backprop_data, in ThenRnnBackward() argument 4716 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward() 4749 DeviceMemory<double> *params_backprop_data, in ThenRnnBackward() argument 4761 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 901 auto params_backprop_data = AsDeviceMemory<T>(params_backprop); in DoBackward() local 917 ¶ms_backprop_data, &reserve_space_uint8, workspace_allocator, in DoBackward()
|