code: def test_force_block_config_elementwise(
      ofm_layout, block_config_str, expected_block_shape
  ):
      op_type = "ethosu_elementwise_unary"
      op_str = "ABS"
      activation = "NONE"
      ofm_shape = (1, 8, 10, 16)
      ifm_matrix = [
          [1, 0, 0, 0, 0],
          [0, 1, 0, 0, 0],
          [0, 0, 1, 0, 0],
          [0, 0, 0, 1, 0],
          [0, 0, 0, 0, 1],
      ]
      ifm_offset = [0, 0, 0, 0]
  
      propagator = cs.Propagator(ifm_matrix, ifm_offset)
  
      op_attrs = {
          "op": op_type,
          "operator_type": op_str,
          "activation": activation,
          "clip_min": 0,
          "clip_max": 0,
          "rounding_mode": "TFL",
      }
  
      config = {
          "enable_cascader": True,
          "dev_force_block_config": block_config_str,
      }
      with tvm.transform.PassContext(
          config={"relay.ext.ethos-u.options": config}
      ):
          device_config = cs.EthosuDeviceConfig("ethos-u55-128")
          block_configs = device_config.get_elementwise_block_config(
              propagator,
              None,
              op_attrs,
              ofm_shape,
              ofm_layout,
              "NWHC",
              None,
              "int8",
              "int8",
          )
  
      assert len(block_configs) == 1
      assert block_configs[0].output_shape == expected_block_shape
  
file_path: tvm/tests/python/contrib/test_ethosu/cascader/test_ethosu_block_config.py
imports: import pytest
  import numpy
  import math
  import tvm
  import tvm.contrib.ethosu.cascader
  from tvm.relay.backend.contrib.ethosu.te.common import get_layout_transform_matrices
  from .infra import make_matrices
label_desc: Use image node_id 3 for calling a global function with example usage: test_force_block_config_elementwise(ofm_layout, block_config_str, expected_block_shape) without return types