Searched defs:grad_scale (Results 1 – 5 of 5) sorted by relevance
488 auto grad_scale = add_binary(feature_shape, HloOpcode::kMultiply, in HandleBatchNormGrad() local
1006 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); in Compile() local
846 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P() local
1079 auto grad_scale = (logbase_bcast * sign_decay_bcast * sign_gm).exp(); in operator ()() local
925 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); in operator ()() local