Method Compute

tensorflow/core/kernels/training_ops.cc:988–1008

  }

  void Compute(OpKernelContext* ctx) override {
    Var* resource;
    const bool sparse = false;
    mutex* mu = GetTrainingVariableMutex<Device, T>(ctx, 0, sparse, &resource);
    core::ScopedUnref scoped_unref(resource);
    if (use_exclusive_lock_ && mu != nullptr) {
      mutex_lock l1(*mu);
      // Don't try to acquire a lock on the second ref as they share the same
      // mutex.
      //
      // mutex_lock l2(*ctx->input_ref_mutex(1));
      DoValidate(ctx);
      if (!ctx->status().ok()) return;
      DoCompute(ctx);
    } else {
      DoValidate(ctx);
      if (!ctx->status().ok()) return;
      DoCompute(ctx);
    }
    MaybeForwardRefInputToRefOutput(ctx, 0, 0);
  }

 private:
  bool use_exclusive_lock_;