class_docstr: Implementation of OneDeviceStrategy.
class_docstr_tok: ["Implementation","of","OneDeviceStrategy","."]
code: def __init__(self, container_strategy, device):
      super(OneDeviceExtended, self).__init__(container_strategy)
      self._device = device_util.resolve(device)
      self._input_device = device_util.get_host_for_device(self._device)
  
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 1 to create a new OneDeviceExtended object from inherited base classes: distribute_lib with example: obj = OneDeviceExtended(container_strategy, device)