Searched refs:offset_grad (Results 1 – 1 of 1) sorted by relevance
1155 void* scale_grad, void* offset_grad, in SetBatchNormBackwardArgs() argument1168 offset_grad, saved_mean, saved_inv_variance); in SetBatchNormBackwardArgs()1599 void* scale_grad, void* offset_grad) { in SetBatchNormBackwardArgs() argument1605 offset_grad, saved_mean, saved_var); in SetBatchNormBackwardArgs()