code: def test_where_fusion():
      """integration test that where and zeros should be properly inlined"""
  
      def check_device(target, dev):
          with tvm.target.Target(target):
              print("Running on target: %s" % target)
              (
                  conv2d_compute,
                  conv2d_schedule,
              ) = tvm.topi.testing.get_conv2d_nchw_implement(target)
              data = te.placeholder((2, 1, 2, 4), "int8", "data")
              w = te.placeholder((3, 1, 2, 2), "int8", "w")
              conv1 = conv2d_compute(data, w, 1, 0, 1, "int32")
              zeros = topi.full(
                  (2, 3, 1, 3), "int32", tvm.tir.const(0, dtype="int32")
              )
              gt = topi.greater_equal(conv1, zeros)
              one = topi.full(
                  (2, 3, 1, 3), "int32", tvm.tir.const(1, dtype="int32")
              )
              two = topi.full(
                  (2, 3, 1, 3), "int32", tvm.tir.const(2, dtype="int32")
              )
              where = topi.where(gt, one, two)
              add = topi.add(conv1, where)
              outs = [add]
              s = conv2d_schedule(outs)
              tvm.build(s, [data, w, add], target=backend)
  
      for backend, dev in tvm.testing.enabled_targets():
          check_device(backend, dev)
  
imports: import numpy
  import pytest
  import tvm
  from tvm import te
  from tvm import topi
  from tvm import relay
  import tvm.topi.testing
  from tvm.contrib.nvcc import have_fp16
  import tvm.testing
label_desc: Use image node_id 53 for calling a global function with example usage: test_where_fusion() without return types