HomeSort by relevance Sort by last modified time
    Searched refs:new_accum (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
training_ops.cc 203 auto new_accum = accum + grad * grad; local
207 grad_with_shrinkage - (new_accum.sqrt() - accum.sqrt()) / lr() * var;
211 (new_accum.pow(-lr_power()) - accum.pow(-lr_power())) / lr() * var;
215 auto y = new_accum.sqrt() / new_accum.constant(lr()) +
222 auto y = new_accum.pow(-lr_power()) / new_accum.constant(lr()) +
242 auto new_accum = accum + grad.square(); local
245 linear.device(d) += grad - (new_accum.sqrt() - accum.sqrt()) / lr() * var;
249 (new_accum.pow(-lr_power()) - accum.pow(-lr_power())) / lr() * var
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 748 // new_accum = accum + grad * grad
750 // (new_accum^(-lr_power) - accum^(-lr_power)) / lr * var
751 // quadratic = (new_accum^(-lr_power) / lr) + 2 * l2
754 // accum = new_accum
764 xla::XlaOp new_accum = accum + xla::Square(grad); local
765 xla::XlaOp new_accum_lr_pow = xla::Pow(new_accum, -lr_power);
    [all...]

Completed in 98 milliseconds