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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 64 xla::XlaOp prox_var = var - grad * lr; local
65 xla::XlaOp l1_gt_zero = xla::Sign(prox_var) *
66 xla::Max(xla::Abs(prox_var) - lr * l1, zero) /
68 xla::XlaOp l1_le_zero = prox_var / (one + lr * l2);
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
training_ops.cc 97 auto prox_var = var; local
99 prox_var.device(d) -= grad * lr();
103 prox_var.sign() *
104 (prox_var.abs() - var.constant(lr() * l1())).cwiseMax(T(0.0)) /
108 prox_var / (var.constant(1.0) + var.constant(l2() * lr()));
175 auto prox_var = var; local
177 prox_var.device(d) -= grad * learning_rate;
180 var.device(d) = prox_var.sign() *
181 (prox_var.abs() - learning_rate * prox_var.constant(l1())
    [all...]

Completed in 119 milliseconds