OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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