HomeSort by relevance Sort by last modified time
    Searched refs:l2_shrinkage (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 394 errors::InvalidArgument("l2_shrinkage is not a scalar: ",
406 xla::ComputationDataHandle l2_shrinkage; local
409 l2_shrinkage = ctx->Input(7);
415 // grad_to_use = grad + 2 * l2_shrinkage * var
427 grad_to_use = b->Add(grad, b->Mul(two, b->Mul(l2_shrinkage, var)));
  /external/tensorflow/tensorflow/core/kernels/
training_ops.h 116 typename TTypes<T>::ConstScalar l2_shrinkage,
training_ops.cc 197 typename TTypes<T>::ConstScalar l2_shrinkage,
199 auto grad_with_shrinkage = grad + static_cast<T>(2) * l2_shrinkage() * var;
1923 const Tensor& l2_shrinkage = ctx->input(7); variable
2087 const Tensor* l2_shrinkage; variable
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 318 milliseconds