code: def test_bf16_build():
      data = relay.var("data", shape=(1, 3, 224, 224), dtype="float32")
      weight = relay.var("weight", shape=(64, 3, 7, 7), dtype="float32")
      bn_gamma = relay.var("gamma", shape=(64,), dtype="float32")
      bn_beta = relay.var("beta", shape=(64,), dtype="float32")
      bn_mean = relay.var("mean", shape=(64,), dtype="float32")
      bn_var = relay.var("var", shape=(64,), dtype="float32")
      params = {
          "weight": np.random.uniform(-1, 1, size=(64, 3, 7, 7)).astype(
              "float32"
          ),
          "gamma": np.random.uniform(-1, 1, size=(64,)).astype(
              "float32"
          ),
          "beta": np.random.uniform(-1, 1, size=(64,)).astype(
              "float32"
          ),
          "mean": np.random.uniform(-1, 1, size=(64,)).astype(
              "float32"
          ),
          "var": np.random.uniform(-1, 1, size=(64,)).astype("float32"),
      }
      conv_bf16 = relay.nn.conv2d(
          relay.cast(data, "bfloat16"),
          relay.cast(weight, "bfloat16"),
          strides=(2, 2),
          padding=(3, 3, 3, 3),
          channels=64,
          kernel_size=(7, 7),
          out_dtype="bfloat16",
      )
      bn_bf16 = relay.nn.batch_norm(
          conv_bf16,
          relay.cast(bn_gamma, "bfloat16"),
          relay.cast(bn_beta, "bfloat16"),
          relay.cast(bn_mean, "bfloat16"),
          relay.cast(bn_var, "bfloat16"),
      )
      relu_bf16 = relay.nn.relu(bn_bf16[0])
      maxpool_bf16 = relay.nn.max_pool2d(
          relu_bf16, pool_size=(2, 2), strides=(2, 2)
      )
      avgpool_bf16 = relay.nn.avg_pool2d(
          maxpool_bf16, pool_size=(2, 2), strides=(2, 2)
      )
      flattened_bf16 = relay.nn.batch_flatten(avgpool_bf16)
      softmax_bf16 = relay.nn.softmax(flattened_bf16)
      mod_bf16 = tvm.IRModule.from_expr(softmax_bf16)
      with tvm.transform.PassContext(opt_level=3):
          relay.build(mod_bf16, target="llvm", params=params)
  
file_path: tvm/tests/python/relay/test_cpp_build_module.py
imports: import numpy
  import tvm
  from tvm import te
  from tvm import relay, runtime
  from tvm.contrib.nvcc import have_fp16
  import tvm.testing
label_desc: Use image node_id 3 for calling a global function with example usage: test_bf16_build() without return types