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.h395 DeviceMemory<double>* backward_bias_data) override;
401 DeviceMemory<float>* backward_bias_data) override;
407 DeviceMemory<Eigen::half>* backward_bias_data) override;
637 DeviceMemory<T>* backward_bias_data);
Dcuda_dnn.cc4008 DeviceMemory<T>* backward_bias_data) { in DoConvolveBackwardBiasImpl() argument
4021 bias_nd.handle(), backward_bias_data->opaque())); in DoConvolveBackwardBiasImpl()
4029 DeviceMemory<double>* backward_bias_data) { in DoConvolveBackwardBias() argument
4032 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
4040 DeviceMemory<float>* backward_bias_data) { in DoConvolveBackwardBias() argument
4043 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
4051 DeviceMemory<Eigen::half>* backward_bias_data) { in DoConvolveBackwardBias() argument
4054 bias_descriptor, backward_bias_data), in DoConvolveBackwardBias()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h419 DeviceMemory<double>* backward_bias_data) override;
425 DeviceMemory<float>* backward_bias_data) override;
431 DeviceMemory<Eigen::half>* backward_bias_data) override;
735 DeviceMemory<T>* backward_bias_data);
Drocm_dnn.cc3758 DeviceMemory<T>* backward_bias_data) { in DoConvolveBackwardBiasImpl() argument
3773 bias_nd.handle(), backward_bias_data->opaque()); in DoConvolveBackwardBiasImpl()
3786 DeviceMemory<double>* backward_bias_data) { in DoConvolveBackwardBias() argument
3795 DeviceMemory<float>* backward_bias_data) { in DoConvolveBackwardBias() argument
3798 backward_bias_data); in DoConvolveBackwardBias()
3805 DeviceMemory<Eigen::half>* backward_bias_data) { in DoConvolveBackwardBias() argument
3808 backward_bias_data); in DoConvolveBackwardBias()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1498 DeviceMemory<float>* backward_bias_data) { in DoConvolveBackwardBias() argument
1506 DeviceMemory<double>* backward_bias_data) { in DoConvolveBackwardBias() argument
1514 DeviceMemory<Eigen::half>* backward_bias_data) { in DoConvolveBackwardBias() argument
Dstream.h502 DeviceMemory<double> *backward_bias_data);
507 DeviceMemory<float> *backward_bias_data);
513 DeviceMemory<Eigen::half> *backward_bias_data);
2097 DeviceMemory<T> *backward_bias_data);
Dstream.cc705 DeviceMemory<T> *backward_bias_data) { in ThenConvolveBackwardBiasImpl() argument
707 PARAM(backward_bias_data)); in ThenConvolveBackwardBiasImpl()
712 backward_bias_data)); in ThenConvolveBackwardBiasImpl()
723 DeviceMemory<double> *backward_bias_data) { in ThenConvolveBackwardBias() argument
725 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()
732 DeviceMemory<float> *backward_bias_data) { in ThenConvolveBackwardBias() argument
734 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()
741 DeviceMemory<Eigen::half> *backward_bias_data) { in ThenConvolveBackwardBias() argument
743 bias_descriptor, backward_bias_data); in ThenConvolveBackwardBias()