code: def init_to_zero(name):
      return lambda nargs: nargs[name].zero_()
  
file_path: triton/python/triton/ops/matmul.py
imports: import torch
  from ..None import Config, autotune, cdiv, heuristics, jit
  from ..None import language
  from .matmul_perf_model import early_config_prune, estimate_matmul_time
label_desc: Use image node_id 3 for calling a global function with example usage: init_to_zero(name) and returns: function