code: def assert_allclose(
      tiny_out: dict, onnx_out: dict, rtol=1e-5, atol=1e-5
  ):
      assert (
          len(tiny_out) == len(onnx_out)
          and tiny_out.keys() == onnx_out.keys()
      )
      for k in tiny_out.keys():
          tiny_v, onnx_v = tiny_out[k], onnx_out[k]
          if tiny_v is None:
              assert tiny_v == onnx_v
          else:
              np.testing.assert_allclose(
                  tiny_v.numpy(),
                  onnx_v,
                  rtol=rtol,
                  atol=atol,
                  err_msg=f"For tensor '{k}' in {tiny_out.keys()}",
              )
  
file_path: tinygrad/test/external/external_model_benchmark.py
imports: import csv, pathlib, time, numpy
  from os import getenv
  import torch
  import onnx
  from onnx.helper import tensor_dtype_to_np_dtype
  import onnxruntime
  from onnx2torch import convert
  from extra.onnx import get_run_onnx
  from tinygrad.helpers import OSX, DEBUG, fetch
  from tinygrad import Tensor, Device
label_desc: Use image node_id 3 for calling a global function with example usage: assert_allclose(tiny_out, onnx_out, rtol, atol) without return types