code: def test_gradient_if():
      x = relay.var("a", shape=(1, 16))
      y = relay.var("y", shape=(1, 16))
      cond = relay.var("cond", shape=(), dtype="uint1")
      net = relay.If(cond, x, x)
      net = relay.add(x, net)
      net = relay.Function([cond, x, y], net)
      mod = tvm.IRModule.from_expr(net)
      mod = relay.transform.ToANormalForm()(mod)
      mod = relay.transform.InferType()(mod)
      mod["main"] = relay.transform.gradient(
          mod["main"], mode="higher_order"
      )
      mod = relay.transform.ToANormalForm()(mod)
  
imports: import pytest
  import sys
  import numpy
  import tvm
  import tvm.testing
  from tvm import te
  from tvm import relay
  from tvm.relay.analysis import detect_feature
  from tvm.relay import op, create_executor, transform
  from tvm.relay.prelude import Prelude
  from tvm.relay.testing import count
  from tvm.relay.analysis import Feature
label_desc: Use image node_id 12 for calling a global function with example usage: test_gradient_if() without return types