Searched refs:x_bn_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 593 dnn::ActivationMode activation_mode, const DeviceMemory<float>& x_bn_data,
609 const DeviceMemory<Eigen::half>& x_bn_data,
749 const DeviceMemory<T>& x_bn_data,
|
| D | rocm_dnn.cc | 4358 const DeviceMemory<T>& x_bn_data, in DoFusedBatchNormActivationBackwardImpl() argument
4396 x_bn_data.opaque(), scale_data.opaque(), offset_data.opaque(), in DoFusedBatchNormActivationBackwardImpl()
4439 const DeviceMemory<float>& x_bn_data, in DoFusedBatchNormActivationBackward() argument
4451 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
4462 const DeviceMemory<Eigen::half>& x_bn_data, in DoFusedBatchNormActivationBackward() argument
4474 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2508 dnn::ActivationMode activation_mode, const DeviceMemory<float>& x_bn_data,
2526 const DeviceMemory<Eigen::half>& x_bn_data,
|