class_docstr: Mesh implementation for TPU using SIMD and MPI operations.
class_docstr_tok: ["Mesh","implementation","for","TPU","using","SIMD","and","MPI","operations","."]
code: def pnum_tensor(self):
      if self._pnum_tensor is not None:
          return self._pnum_tensor
      with utils.outside_all_rewrites():
          tf.logging.info("Create pnum_tensor")
          self._pnum_tensor = tpu_ops.tpu_replicated_input(
              self._physical_to_logical, name="pnum_constants"
          )
          return self._pnum_tensor
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import math
  import os
  import gin
  from mesh_tensorflow import ops_with_redefined_builtins
  from mesh_tensorflow import tpu_variables
  from mesh_tensorflow import utils
  from six.moves import xrange
  import tensorflow.compat.v1
  from tensorflow.python.tpu.ops import tpu_ops
label_desc: Use image node_id 2 for calling the SimdMeshImpl obj's underlying member method code with example usage: obj.pnum_tensor() and returns: self, self