code: def test_dfg():
      """Tests dataflow graph"""
      A = te.placeholder((1024, 4096), dtype="float32", name="A")
      B = topi.nn.softmax(A)
      # confirm lower works
      s = te.create_schedule([B.op])
  
      def verify():
          # pylint: disable=import-outside-toplevel
          from tvm.contrib import tedd
  
          _str = tedd.viz_dataflow_graph(s, False, "", True)
          # Check all edges are available
          findany(r"digraph \"Dataflow Graph\"", str)
          findany(r"Stage_0:O_0 -> Tensor_0_0", str)
          findany(r"Tensor_0_0 -> Stage_1:I_0", str)
          findany(r"Stage_1:O_0 -> Tensor_1_0", str)
          findany(r"Tensor_0_0 -> Stage_2:I_0", str)
          findany(r"Tensor_1_0 -> Stage_2:I_1", str)
          findany(r"Stage_2:O_0 -> Tensor_2_0", str)
          findany(r"Tensor_2_0 -> Stage_3:I_0", str)
          findany(r"Stage_3:O_0 -> Tensor_3_0", str)
          findany(r"Tensor_2_0 -> Stage_4:I_0", str)
          findany(r"Tensor_3_0 -> Stage_4:I_1", str)
          findany(r"Stage_4:O_0 -> Tensor_4_0", str)
  
      if checkdependency():
          verify()
  
file_path: tvm/tests/python/contrib/test_tedd.py
imports: import re
  import tvm
  from tvm import te
  from tvm import topi
  from tvm import relay
  from tvm.relay import testing
  from tvm.relay.backend import Runtime, Executor
label_desc: Use image node_id 3 for calling a global function with example usage: test_dfg() without return types