class_docstr: Implementation of OneDeviceStrategy.
class_docstr_tok: ["Implementation","of","OneDeviceStrategy","."]
code: def _update_non_slot(self, colocate_with, fn, args, kwargs, group):
      del colocate_with
      with ops.device(self._device), distribute_lib.UpdateContext(
          self._device
      ):
          result = fn(*args, **kwargs)
          if group:
              return result
          else:
              return nest.map_structure(self._local_results, result)
  
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 18 for calling the OneDeviceExtended obj's underlying member method code with example usage: obj._update_non_slot(colocate_with, fn, args, kwargs, group) and returns: result, nest