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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc 120 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_nesterov", &use_nesterov_));
157 if (use_nesterov_) {
169 bool use_nesterov_; member in class:tensorflow::__anon44191::ResourceApplyMomentum
178 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_nesterov", &use_nesterov_));
215 if (use_nesterov_) {
227 bool use_nesterov_; member in class:tensorflow::__anon44191::ResourceApplyKerasMomentum
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
training_ops.cc 2424 bool use_nesterov_; member in class:tensorflow::ApplyMomentumOp
2562 bool use_nesterov_; member in class:tensorflow::SparseApplyMomentumOp
2645 bool use_nesterov_; member in class:tensorflow::ApplyKerasMomentumOp
2780 bool use_nesterov_; member in class:tensorflow::SparseApplyKerasMomentumOp
2889 bool use_nesterov_; member in class:tensorflow::ApplyAdamOp
    [all...]

Completed in 308 milliseconds