code: def single_elementwise(
      A: T.Buffer((128, 128), "float32"),
      B: T.Buffer((128, 128), "float32"),
  ):
      for i, j in T.grid(128, 128):
          with T.block("B"):
              vi, vj = T.axis.remap("SS", [i, j])
              B[vi, vj] = A[vi, vj] * 2.0
  
file_path: tvm/tests/python/tir-schedule/test_tir_schedule_blockize.py
imports: import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.testing import verify_trace_roundtrip
  import pytest
label_desc: Use image node_id 1 for calling a global function with example usage: single_elementwise(A, B) without return types