HomeSort by relevance Sort by last modified time
    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