code: def test_red_gray_propagates_simple(target_precision):
      """Everything after a softmax should be in FP32 (exception green colored ops)"""
      shape = [1, 2, 3]
      a = relay.var("a", shape=shape)
      b = relay.nn.softmax(a)
      c = b + b
      mod = tvm.IRModule.from_expr(c)
      mod = tvm.relay.transform.InferType()(mod)
  
      mod_params = {
          "a": np.random.uniform(-1, 1, size=shape).astype("float32"),
      }
      output_mod = verify_mixed_precision_output_close(
          mod,
          mod_params,
          mixed_precision_dtype=target_precision,
          atol=0.0,
          rtol=0.0,
      )
  
      assert tvm.ir.structural_equal(mod, output_mod)
  
imports: from typing import Any, Dict, List
  import numpy
  import pytest
  import tvm
  from tvm import relay
  from tvm.relay.testing import lstm
  from tvm.relay.transform import InferType, ToMixedPrecision, mixed_precision
label_desc: Use image node_id 13 for calling a global function with example usage: test_red_gray_propagates_simple(target_precision) without return types