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 726 errors::InvalidArgument("l2_shrinkage is not a scalar: ",
738 xla::XlaOp l2_shrinkage; local
741 l2_shrinkage = ctx->Input(7);
747 // grad_to_use = grad + 2 * l2_shrinkage * var
759 grad_to_use = grad + two * l2_shrinkage * var;
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
training_ops.h 116 typename TTypes<T>::ConstScalar l2_shrinkage,
training_ops.cc 200 typename TTypes<T>::ConstScalar l2_shrinkage,
202 auto grad_with_shrinkage = grad + static_cast<T>(2) * l2_shrinkage() * var;
2022 const Tensor& l2_shrinkage = ctx->input(7); variable
2189 const Tensor* l2_shrinkage; variable
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 377 milliseconds