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

  /external/tensorflow/tensorflow/core/kernels/
training_ops.cc 199 auto grad_with_shrinkage = grad + static_cast<T>(2) * l2_shrinkage() * var; local
200 auto new_accum = accum + grad_with_shrinkage.square();
204 grad_with_shrinkage - (new_accum.sqrt() - accum.sqrt()) / lr() * var;
207 grad_with_shrinkage -
225 accum.device(d) += grad_with_shrinkage.square();
2153 auto grad_with_shrinkage = variable
2155 COMPUTE_FTRL(grad_with_shrinkage); variable
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 477 milliseconds