Searched refs:copy_on_read_mode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_op_helpers.h | 33 if (var->copy_on_read_mode.load()) { in EnsureSparseVariableAccess()
41 var->copy_on_read_mode.store(true); in EnsureSparseVariableAccess()
68 var->copy_on_read_mode.store(true); in EnsureSparseVariableAccess()
199 bool copy_on_read_mode) { in PrepareToUpdateVariable() argument
200 if (copy_on_read_mode || !tensor->RefCountIsOne()) { in PrepareToUpdateVariable()
251 ctx, var->tensor(), var->copy_on_read_mode.load())); in GetInputTensorFromVariable()
|
| D | resource_variable_ops.cc | 150 if (!variable->copy_on_read_mode.load()) { in Compute()
205 if (variables[i]->copy_on_read_mode.load()) { in Compute()
393 if (variable->copy_on_read_mode.load()) { in Compute()
543 context, var_tensor, variable->copy_on_read_mode.load())); in Compute()
|
| D | stateful_random_ops.cc | 111 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
284 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
|
| D | random_binomial_op.cc | 422 ctx, var_tensor, var->copy_on_read_mode.load())); in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_var.h | 88 std::atomic<bool> copy_on_read_mode{false};
|