Searched defs:ComputeGradientScale (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| H A D | training_ops.cc | 1050 xla::XlaOp ComputeGradientScale(xla::XlaOp alpha, xla::XlaOp decay) override { in ComputeGradientScale() function in tensorflow::__anon7a3fe8190111::ResourceApplyAddSign |