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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 416 // new_accum = accum + grad_to_use * grad_to_use
418 // (new_accum^(-lr_power) - accum^(-lr_power)) / lr * var
419 // quadratic = (new_accum^(-lr_power) / lr) + 2 * l2
422 // accum = new_accum
432 xla::ComputationDataHandle new_accum = local
435 b->Pow(new_accum, b->Neg(lr_power));
445 accum = new_accum;
  /external/tensorflow/tensorflow/core/kernels/
training_ops.cc 200 auto new_accum = accum + grad_with_shrinkage.square(); local
204 grad_with_shrinkage - (new_accum.sqrt() - accum.sqrt()) / lr() * var;
208 (new_accum.pow(-lr_power()) - accum.pow(-lr_power())) / lr() * var;
212 auto y = new_accum.sqrt() / new_accum.constant(lr()) +
219 auto y = new_accum.pow(-lr_power()) / new_accum.constant(lr()) +
239 auto new_accum = accum + grad.square(); local
242 linear.device(d) += grad - (new_accum.sqrt() - accum.sqrt()) / lr() * var;
246 (new_accum.pow(-lr_power()) - accum.pow(-lr_power())) / lr() * var
    [all...]

Completed in 116 milliseconds