Searched refs:scale_grad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1042 void* scale_grad, void* offset_grad, in SetBatchNormBackwardArgs() argument
1054 fusion_args_, batchnorm_op, alpha, beta, x, scale, offset, scale_grad, in SetBatchNormBackwardArgs()
1486 void* scale_grad, void* offset_grad) { in SetBatchNormBackwardArgs() argument
1491 k_batchnorm_op_idx, &alpha, &beta, x, scale, offset, scale_grad, in SetBatchNormBackwardArgs()
|