Searched refs:offset_grad (Results 1 – 1 of 1) sorted by relevance
1042 void* scale_grad, void* offset_grad, in SetBatchNormBackwardArgs() argument1055 offset_grad, saved_mean, saved_inv_variance); in SetBatchNormBackwardArgs()1486 void* scale_grad, void* offset_grad) { in SetBatchNormBackwardArgs() argument1492 offset_grad, saved_mean, saved_var); in SetBatchNormBackwardArgs()