code: def _old_tf_stateless_truncated_normal(
      shape,
      seed,
      mean=0.0,
      stddev=1.0,
      dtype=dtypes.float32,
      name=None,
      layout=None,
  ):
      """DTensor stateless truncated normal implementation that takes an layout."""
      with ops.name_scope(
          name,
          "stateless_truncated_normal",
          [shape, seed, mean, stddev],
      ) as name:
          seed = ops.convert_to_tensor(
              seed, dtype=dtypes.int32, name="seed"
          )
          shape = shape_util.shape_tensor(shape)
          mean = ops.convert_to_tensor(mean, dtype=dtype, name="mean")
          stddev = ops.convert_to_tensor(
              stddev, dtype=dtype, name="stddev"
          )
          rnd = api.call_with_layout(
              gen_stateless_random_ops.stateless_truncated_normal,
              layout,
              shape,
              seed,
              dtype,
          )
          result = math_ops.add(rnd * stddev, mean, name=name)
          shape_util.maybe_set_static_shape(result, shape)
          return result
  
file_path: tensorflow/tensorflow/dtensor/python/d_random.py
imports: from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import layout
  from tensorflow.python.eager import context
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gen_math_ops
  from tensorflow.python.ops import gen_stateless_random_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import shape_util
label_desc: Use image node_id 3 for calling a global function with example usage: _old_tf_stateless_truncated_normal(shape, seed, mean, stddev, dtype, name, layout) and returns: result