Searched refs:grad_beta (Results 1 – 2 of 2) sorted by relevance
534 auto grad_beta = add(HloInstruction::CreateReduce( in HandleBatchNormGrad() local544 auto i2 = add(HloInstruction::CreateBroadcast(activation_shape, grad_beta, in HandleBatchNormGrad()582 HloInstruction::CreateTuple({grad_activation, grad_scale, grad_beta}); in HandleBatchNormGrad()
267 grad_alpha, grad_beta = tape.gradient(samples, [alpha, beta])269 self.assertIsNotNone(grad_beta)