OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:lr_power
(Results
1 - 5
of
5
) sorted by null
/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/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
...]
Completed in 386 milliseconds