Home
last modified time | relevance | path

Searched refs:backward_bias_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h376 DeviceMemory<double>* backward_bias_data) override;
382 DeviceMemory<float>* backward_bias_data) override;
388 DeviceMemory<Eigen::half>* backward_bias_data) override;
603 DeviceMemory<T>* backward_bias_data);
Dcuda_dnn.cc3475 DeviceMemory<T>* backward_bias_data) { in DoConvolveBackwardBiasImpl() argument
3488 bias_nd.handle(), backward_bias_data->opaque())); in DoConvolveBackwardBiasImpl()
3496 DeviceMemory<double>* backward_bias_data) { in DoConvolveBackwardBias() argument
3499 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
3507 DeviceMemory<float>* backward_bias_data) { in DoConvolveBackwardBias() argument
3510 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
3518 DeviceMemory<Eigen::half>* backward_bias_data) { in DoConvolveBackwardBias() argument
3521 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h367 DeviceMemory<double>* backward_bias_data) override;
373 DeviceMemory<float>* backward_bias_data) override;
379 DeviceMemory<Eigen::half>* backward_bias_data) override;
660 DeviceMemory<T>* backward_bias_data);
Drocm_dnn.cc3232 DeviceMemory<T>* backward_bias_data) { in DoConvolveBackwardBiasImpl() argument
3247 bias_nd.handle(), backward_bias_data->opaque()); in DoConvolveBackwardBiasImpl()
3260 DeviceMemory<double>* backward_bias_data) { in DoConvolveBackwardBias() argument
3269 DeviceMemory<float>* backward_bias_data) { in DoConvolveBackwardBias() argument
3272 backward_bias_data); in DoConvolveBackwardBias()
3279 DeviceMemory<Eigen::half>* backward_bias_data) { in DoConvolveBackwardBias() argument
3282 backward_bias_data); in DoConvolveBackwardBias()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1432 DeviceMemory<float>* backward_bias_data) {
1440 DeviceMemory<double>* backward_bias_data) {
1448 DeviceMemory<Eigen::half>* backward_bias_data) {
Dstream.h479 DeviceMemory<double> *backward_bias_data);
484 DeviceMemory<float> *backward_bias_data);
490 DeviceMemory<Eigen::half> *backward_bias_data);
2030 DeviceMemory<T> *backward_bias_data);
Dstream.cc1096 DeviceMemory<T> *backward_bias_data) { in ThenConvolveBackwardBiasImpl() argument
1098 PARAM(backward_bias_data)); in ThenConvolveBackwardBiasImpl()
1104 backward_bias_data)); in ThenConvolveBackwardBiasImpl()
1116 DeviceMemory<double> *backward_bias_data) { in ThenConvolveBackwardBias() argument
1118 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()
1125 DeviceMemory<float> *backward_bias_data) { in ThenConvolveBackwardBias() argument
1127 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()
1134 DeviceMemory<Eigen::half> *backward_bias_data) { in ThenConvolveBackwardBias() argument
1136 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()