Searched refs:y_act_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 592 const DeviceMemory<float>& y_act_data,
607 const DeviceMemory<Eigen::half>& y_act_data,
748 const DeviceMemory<T>& y_act_data, dnn::ActivationMode activation_mode,
|
| D | rocm_dnn.cc | 4357 const DeviceMemory<T>& y_act_data, dnn::ActivationMode activation_mode, in DoFusedBatchNormActivationBackwardImpl() argument
4403 y_act_data.opaque()); in DoFusedBatchNormActivationBackwardImpl()
4438 const DeviceMemory<float>& y_act_data, dnn::ActivationMode activation_mode, in DoFusedBatchNormActivationBackward() argument
4451 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
4460 const DeviceMemory<Eigen::half>& y_act_data, in DoFusedBatchNormActivationBackward() argument
4474 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2507 const DeviceMemory<float>& y_act_data,
2524 const DeviceMemory<Eigen::half>& y_act_data,
|