code: def apply_layout_transform(
      data: relay.Expr, src_layout: str, dst_layout: str
  ):
      assert (
          relay.op.get("layout_transform").get_attr("TOpPattern")
          == OpPattern.INJECTIVE
      )
      return relay.layout_transform(data, src_layout, dst_layout)
  
imports: import itertools
  import random
  import tempfile
  from typing import Callable, Dict, List, Optional, Tuple, Union
  import numpy
  import pytest
  import tvm
  import tvm.testing
  from tvm import meta_schedule, relay
  from tvm.meta_schedule.schedule.cuda.layout_transform import cuda_layout_transform_schedule_rule
  from tvm.relay.op import OpPattern
  from tvm.script import ir
  from tvm.script import tir
  from tvm.tir.schedule import BlockRV
label_desc: Use image node_id 6 for calling a global function with example usage: apply_layout_transform(data, src_layout, dst_layout) and returns: relay