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