Searched refs:copy_on_read_mode (Results 1 – 5 of 5) sorted by relevance
32 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() argument201 if (copy_on_read_mode || !tensor->RefCountIsOne()) { in PrepareToUpdateVariable()253 ctx, var->tensor(), var->copy_on_read_mode.load())); in GetInputTensorFromVariable()
151 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()
90 ctx, var_tensor, var->copy_on_read_mode.load())); in UpdateVariableAndFill()
88 std::atomic<bool> copy_on_read_mode{false};
625 bool copy_on_read_mode); // For access to RefCountIsOne().