Searched refs:y_act_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 644 const DeviceMemory<float>& y_act_data,
659 const DeviceMemory<Eigen::half>& y_act_data,
823 const DeviceMemory<T>& y_act_data, dnn::ActivationMode activation_mode,
|
| D | rocm_dnn.cc | 4953 const DeviceMemory<T>& y_act_data, dnn::ActivationMode activation_mode, in DoFusedBatchNormActivationBackwardImpl() argument
4999 y_act_data.opaque()); in DoFusedBatchNormActivationBackwardImpl()
5034 const DeviceMemory<float>& y_act_data, dnn::ActivationMode activation_mode, in DoFusedBatchNormActivationBackward() argument
5047 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
5056 const DeviceMemory<Eigen::half>& y_act_data, in DoFusedBatchNormActivationBackward() argument
5070 y_act_data, activation_mode, x_bn_data, in DoFusedBatchNormActivationBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2642 const DeviceMemory<float>& y_act_data, in DoFusedBatchNormActivationBackward() argument
2659 const DeviceMemory<Eigen::half>& y_act_data, in DoFusedBatchNormActivationBackward() argument
|