Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h147 DeviceMemory<Eigen::half>* params_backprop_data,
172 DeviceMemory<float>* params_backprop_data,
197 DeviceMemory<double>* params_backprop_data,
645 DeviceMemory<T>* params_backprop_data,
Dcuda_dnn.cc1679 DeviceMemory<T>* params_backprop_data, in DoRnnBackwardImpl() argument
1763 if (params_backprop_data != nullptr) { in DoRnnBackwardImpl()
1765 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
1777 /*dw=*/params_backprop_data->opaque(), in DoRnnBackwardImpl()
1796 /*dw=*/params_backprop_data->opaque(), in DoRnnBackwardImpl()
2017 DeviceMemory<Eigen::half>* params_backprop_data, in DoRnnBackward() argument
2043 params_backprop_data, reserve_space_data, workspace_allocator, in DoRnnBackward()
2068 DeviceMemory<float>* params_backprop_data, in DoRnnBackward() argument
2094 params_backprop_data, reserve_space_data, workspace_allocator, in DoRnnBackward()
2120 DeviceMemory<double>* params_backprop_data, in DoRnnBackward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h138 DeviceMemory<Eigen::half>* params_backprop_data,
163 DeviceMemory<float>* params_backprop_data,
188 DeviceMemory<double>* params_backprop_data,
700 DeviceMemory<T>* params_backprop_data,
Drocm_dnn.cc2136 DeviceMemory<T>* params_backprop_data, in DoRnnBackwardImpl() argument
2209 if (params_backprop_data != nullptr) { in DoRnnBackwardImpl()
2211 stream->ThenMemZero(params_backprop_data, params_backprop_data->size()); in DoRnnBackwardImpl()
2219 params_backprop_data->opaque() /*dw*/, workspace.opaque() /*workspace*/, in DoRnnBackwardImpl()
2452 DeviceMemory<Eigen::half>* params_backprop_data, in DoRnnBackward() argument
2479 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2503 DeviceMemory<float>* params_backprop_data, in DoRnnBackward() argument
2530 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2555 DeviceMemory<double>* params_backprop_data, in DoRnnBackward() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2240 DeviceMemory<Eigen::half>* params_backprop_data,
2268 DeviceMemory<float>* params_backprop_data,
2296 DeviceMemory<double>* params_backprop_data,
Dstream.h1815 DeviceMemory<Eigen::half> *params_backprop_data,
1840 DeviceMemory<float> *params_backprop_data,
1865 DeviceMemory<double> *params_backprop_data,
Dstream.cc4976 DeviceMemory<Eigen::half> *params_backprop_data, in ThenRnnBackward() argument
4989 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
5022 DeviceMemory<float> *params_backprop_data, in ThenRnnBackward() argument
5035 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
5069 DeviceMemory<double> *params_backprop_data, in ThenRnnBackward() argument
5082 params_backprop_data, reserve_space_data, workspace_allocator, in ThenRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc841 auto params_backprop_data = AsDeviceMemory<T>(params_backprop); in DoBackward() local
857 &params_backprop_data, &reserve_space_uint8, in DoBackward()