class_docstr: Implementation of OneDeviceStrategy.
class_docstr_tok: ["Implementation","of","OneDeviceStrategy","."]
code: def _update(self, var, fn, args, kwargs, group):
      # The implementations of _update() and _update_non_slot() are identical
      # except _update() passes `var` as the first argument to `fn()`.
      return self._update_non_slot(
          var, fn, (var,) + tuple(args), kwargs, group
      )
  
imports: from tensorflow.python.distribute import device_util
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import distribute_utils
  from tensorflow.python.distribute import input_lib
  from tensorflow.python.distribute import input_util
  from tensorflow.python.distribute import numpy_dataset
  from tensorflow.python.distribute.v1 import input_lib
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.util import nest
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 17 for calling the OneDeviceExtended obj's underlying member method code with example usage: obj._update(var, fn, args, kwargs, group) and returns: self