Home
last modified time | relevance | path

Searched refs:copy_on_read_mode (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtraining_op_helpers.h32 if (var->copy_on_read_mode.load()) { in EnsureSparseVariableAccess()
40 var->copy_on_read_mode.store(true); in EnsureSparseVariableAccess()
67 var->copy_on_read_mode.store(true); in EnsureSparseVariableAccess()
200 bool copy_on_read_mode) { in PrepareToUpdateVariable() argument
201 if (copy_on_read_mode || !tensor->RefCountIsOne()) { in PrepareToUpdateVariable()
253 ctx, var->tensor(), var->copy_on_read_mode.load())); in GetInputTensorFromVariable()
Dresource_variable_ops.cc151 if (variable->copy_on_read_mode.load()) { in Compute()
202 if (variables[i]->copy_on_read_mode.load()) { in Compute()
362 if (variable->copy_on_read_mode.load()) { in Compute()
513 context, var_tensor, variable->copy_on_read_mode.load())); in Compute()
Dstateful_random_ops.cc90 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
/external/tensorflow/tensorflow/core/framework/
Dresource_var.h88 std::atomic<bool> copy_on_read_mode{false};
Dtensor.h625 bool copy_on_read_mode); // For access to RefCountIsOne().