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.h148 DeviceMemory<Eigen::half>* params_backprop_data,
173 DeviceMemory<float>* params_backprop_data,
198 DeviceMemory<double>* params_backprop_data,
679 DeviceMemory<T>* params_backprop_data,
Dcuda_dnn.cc1906 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/
Drocm_dnn.h175 DeviceMemory<Eigen::half>* params_backprop_data,
200 DeviceMemory<float>* params_backprop_data,
225 DeviceMemory<double>* params_backprop_data,
775 DeviceMemory<T>* params_backprop_data,
Drocm_dnn.cc2217 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/
Ddnn.h2309 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
Dstream.h1857 DeviceMemory<Eigen::half> *params_backprop_data,
1882 DeviceMemory<float> *params_backprop_data,
1907 DeviceMemory<double> *params_backprop_data,
Dstream.cc4660 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/
Dcudnn_rnn_ops.cc901 auto params_backprop_data = AsDeviceMemory<T>(params_backprop); in DoBackward() local
917 &params_backprop_data, &reserve_space_uint8, workspace_allocator, in DoBackward()