code: def get_layout_optimizer(self):
      return layout_optimizer.LayoutOptimizer(
          memory_estimator.MemoryEstimator(
              self.mtf_graph, self.mesh_shape
          )
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import mesh_tensorflow
  from mesh_tensorflow.auto_mtf import layout_optimizer
  from mesh_tensorflow.auto_mtf import memory_estimator
  import six
  import tensorflow.compat.v1
label_desc: Use image node_id 2 for calling the OptimizeLayoutTest obj's underlying member method code with example usage: obj.get_layout_optimizer() and returns: layout_optimizer