code: def test_tflite_matmul(
      accel_type,
      ifm_shape,
      ofm_channels,
      activation_function,
  ):
      np.random.seed(0)
  
      @tf.function
      def matmul(x, y):
          x = tf.matmul(x, y, transpose_b=True)
          if activation_function == "RELU":
              x = tf.nn.relu(x)
          return x
  
      infra.compare_tvm_with_tflite(
          matmul,
          [ifm_shape, [ofm_channels, ifm_shape[-1]]],
          accel_type,
          enable_cascader=False,
      )
  
imports: import pytest
  import numpy
  import tflite.Model
  import tvm
  import tensorflow
  from tvm import relay
  from tvm.relay.backend.contrib.ethosu import util
  from tvm.relay.op.contrib.ethosu import partition_for_ethosu
  from tvm.testing.aot import generate_ref_data
  from .None import infra
label_desc: Use image node_id 50 for calling a global function with example usage: test_tflite_matmul(accel_type, ifm_shape, ofm_channels, activation_function) and returns: x