code: def _apply_desired_layout_simd(relay_mod):
      desired_layouts = {
          "qnn.conv2d": ["NHWC", "HWOI"],
          "nn.conv2d": ["NHWC", "HWOI"],
      }
  
      seq = tvm.transform.Sequential(
          [
              relay.transform.RemoveUnusedFunctions(),
              relay.transform.ConvertLayout(desired_layouts),
          ]
      )
  
      with tvm.transform.PassContext(opt_level=3):
          return seq(relay_mod)
  
file_path: tvm/tests/micro/zephyr/test_zephyr_armv7m.py
imports: import os
  import pathlib
  import pytest
  import numpy
  import tvm
  import tvm.rpc
  import tvm.micro
  import tvm.testing
  from tvm import relay
  from tvm.contrib.download import download_testdata
  from tvm.relay.backend import Executor, Runtime
  from .None import utils
label_desc: Use image node_id 3 for calling a global function with example usage: _apply_desired_layout_simd(relay_mod) and returns: seq