Home
last modified time | relevance | path

Searched refs:use_exclusive_lock_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops.cc936 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_locking", &use_exclusive_lock_)); in ApplyGradientDescentOp()
942 ctx, use_exclusive_lock_, sparse, {0}); in Compute()
945 ctx, 0, use_exclusive_lock_, sparse, &var)); in Compute()
970 bool use_exclusive_lock_; member in tensorflow::ApplyGradientDescentOp
1023 OP_REQUIRES_OK(ctx, ctx->GetAttr("use_locking", &use_exclusive_lock_)); in ApplyAdadeltaOp()
1029 ctx, use_exclusive_lock_, sparse, {0, 1, 2}); in Compute()
1037 bool use_exclusive_lock_; member in tensorflow::ApplyAdadeltaOp
1043 ctx, 0, use_exclusive_lock_, sparse, &var)); in DoValidate()
1046 ctx, 1, use_exclusive_lock_, sparse, &accum)); in DoValidate()
1049 ctx, GetInputTensorFromVariable<Device, T>(ctx, 2, use_exclusive_lock_, in DoValidate()
[all …]
Dassign_op.h34 context->GetAttr("use_locking", &use_exclusive_lock_)); in AssignOp()
96 if (use_exclusive_lock_) { in Compute()
122 if (use_exclusive_lock_) { in Compute()
139 bool use_exclusive_lock_; variable
Ddense_update_ops.cc50 context->GetAttr("use_locking", &use_exclusive_lock_)); in DenseUpdateOp()
60 if (use_exclusive_lock_) { in Compute()
70 Tensor Tparams = context->mutable_input(0, use_exclusive_lock_); in DoUpdate()
85 bool use_exclusive_lock_; member in tensorflow::DenseUpdateOp
Dscatter_op.cc75 OP_REQUIRES_OK(c, c->GetAttr("use_locking", &use_exclusive_lock_)); in ScatterUpdateOp()
79 if (use_exclusive_lock_) { in Compute()
89 bool use_exclusive_lock_; member in tensorflow::ScatterUpdateOp
92 Tensor params = c->mutable_input(0, use_exclusive_lock_); in DoCompute()
Dscatter_nd_op.cc245 OP_REQUIRES_OK(c, c->GetAttr("use_locking", &use_exclusive_lock_)); in ScatterNdUpdateOp()
248 use_exclusive_lock_ = false; in ScatterNdUpdateOp()
259 } else if (use_exclusive_lock_) { in Compute()
272 bool use_exclusive_lock_; member in tensorflow::ScatterNdUpdateOp
287 params = c->mutable_input(0, use_exclusive_lock_); in DoCompute()
Dresource_variable_ops.cc843 Status s = c->GetAttr("use_locking", &use_exclusive_lock_); in ResourceScatterUpdateOp()
845 use_exclusive_lock_ = false; in ResourceScatterUpdateOp()
856 if (is_non_pod_dtype || use_exclusive_lock_) { in Compute()
867 bool use_exclusive_lock_; member in tensorflow::ResourceScatterUpdateOp