class_docstr: Creates mtf graph, mesh, and mesh implementation.
class_docstr_tok: ["Creates","mtf","graph",",","mesh",",","and","mesh","implementation","."]
code: def device_assignment(self):
      return self._d_assignment
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import re
  import mesh_tensorflow
  import numpy
  import six
  from six.moves import range
  from six.moves import zip
  import tensorflow.compat.v1
  from tensorflow.compat.v1 import estimator
  from mesh_tensorflow.experimental import input_reader
  from mesh_tensorflow.experimental import unet
  from tensorflow.contrib import summary
  from tensorflow.contrib import tpu
  from tensorflow.contrib.tpu.python.tpu import device_assignment
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import control_flow_ops
  from tensorflow.python.platform import flags
  from tensorflow.python.tpu.ops import tpu_ops
label_desc: Use image node_id 3 for calling the MeshContext obj's underlying member method code with example usage: obj.device_assignment() and returns: self