Searched refs:ComputeGradientScale (Results 1 – 1 of 1) sorted by relevance
1006 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); in Compile()1030 virtual xla::XlaOp ComputeGradientScale(xla::XlaOp alpha,1050 xla::XlaOp ComputeGradientScale(xla::XlaOp alpha, xla::XlaOp decay) override { in ComputeGradientScale() function in tensorflow::__anon7a3fe8190111::ResourceApplyAddSign1070 xla::XlaOp ComputeGradientScale(xla::XlaOp alpha, xla::XlaOp decay) override { in ComputeGradientScale() function in tensorflow::__anon7a3fe8190111::ResourceApplyPowerSign