| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| api_def_ApplyFtrl.pbtxt | 46 name: "lr_power"
68 linear += grad + (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
69 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_ApplyFtrlV2.pbtxt | 46 name: "lr_power"
70 (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
71 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_ResourceApplyFtrl.pbtxt | 46 name: "lr_power"
62 linear += grad - (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
63 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_ResourceApplyFtrlV2.pbtxt | 46 name: "lr_power"
64 (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
65 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_ResourceSparseApplyFtrl.pbtxt | 52 name: "lr_power"
69 linear += grad + (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
70 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_ResourceSparseApplyFtrlV2.pbtxt | 52 name: "lr_power"
71 (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
72 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_SparseApplyFtrl.pbtxt | 52 name: "lr_power"
75 linear += grad + (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
76 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| api_def_SparseApplyFtrlV2.pbtxt | 52 name: "lr_power"
77 (accum_new^(-lr_power) - accum^(-lr_power)) / lr * var
78 quadratic = 1.0 / (accum_new^(lr_power) * lr) + 2 * l2
|
| /external/tensorflow/tensorflow/python/training/ |
| training_ops_test.py | 94 lr_power=-0.5):
104 lr_power)
108 linear_update = z + grad - (accum_update**(-lr_power) - y**
109 (-lr_power)) / lr * x
110 quadratic = 1.0 / (accum_update**(lr_power) * lr) + 2 * l2
178 lr_power=-0.5):
196 lr_power=lr_power)
203 (lr_power), var.eval()[index])
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| training_ops.cc | 399 errors::InvalidArgument("lr_power is not a scalar: ",
407 xla::ComputationDataHandle lr_power; local
410 lr_power = ctx->Input(8);
412 lr_power = ctx->Input(7);
418 // (new_accum^(-lr_power) - accum^(-lr_power)) / lr * var
419 // quadratic = (new_accum^(-lr_power) / lr) + 2 * l2
435 b->Pow(new_accum, b->Neg(lr_power));
436 xla::ComputationDataHandle accum_lr_pow = b->Pow(accum, b->Neg(lr_power));
|
| /external/tensorflow/tensorflow/core/ops/ |
| training_ops.cc | 468 TF_RETURN_IF_ERROR(c->WithRank(c->input(idx++), 0, &unused)); // lr_power
483 .Input("lr_power: T")
500 .Input("lr_power: T")
517 .Input("lr_power: T")
533 .Input("lr_power: T")
550 .Input("lr_power: T")
568 .Input("lr_power: T")
586 .Input("lr_power: T")
603 .Input("lr_power: T")
|
| training_ops_test.cc | 206 // lr, l1, l2, and lr_power must be scalars.
229 // lr, l1, l2, and lr_power must be scalars.
|
| ops.pbtxt | 1165 name: "lr_power"
1244 name: "lr_power"
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| training_ops.h | 104 typename TTypes<T>::ConstScalar lr_power);
117 typename TTypes<T>::ConstScalar lr_power);
|
| training_ops.cc | 198 typename TTypes<T>::ConstScalar lr_power) {
201 // special case for which lr_power=-0.5.
202 if (lr_power() == static_cast<T>(-0.5)) {
208 (new_accum.pow(-lr_power()) - accum.pow(-lr_power())) / lr() * var;
211 if (lr_power() == static_cast<T>(-0.5)) {
219 auto y = new_accum.pow(-lr_power()) / new_accum.constant(lr()) +
238 typename TTypes<T>::ConstScalar lr_power) {
240 // special case for which lr_power=-0.5.
241 if (lr_power() == static_cast<T>(-0.5))
1913 const Tensor& lr_power = ctx->input(lr_power_index); variable
2063 const Tensor& lr_power = ctx->input(lr_power_index); variable
[all...] |
| /external/tensorflow/tensorflow/go/op/ |
| wrappers.go | [all...] |
| /external/tensorflow/tensorflow/core/ops/compat/ |
| ops_history.v0.pbtxt | 1667 name: "lr_power"
1737 name: "lr_power"
1808 name: "lr_power"
[all...] |
| ops_history.v1.pbtxt | [all...] |
