code: def residual_store_ds_implementation(
      vals, bias, res, gamma, beta, epsilon
  ):
      global inference_module
      if inference_module is None:
          inference_module = InferenceBuilder().load()
      return inference_module.layer_norm_residual_store_pre_ln_res(
          vals, bias, res, gamma, beta, epsilon
      )
  
file_path: DeepSpeed/tests/unit/ops/transformer/inference/test_layer_norm.py
imports: import deepspeed
  import torch
  import pytest
  from deepspeed.accelerator import get_accelerator
  from deepspeed.ops.op_builder import InferenceBuilder
  from .inference_test_utils import allclose, get_dtypes, assert_almost_equal
label_desc: Use image node_id 10 for calling a global function with example usage: residual_store_ds_implementation(vals, bias, res, gamma, beta, epsilon) and returns: inference_module