Searched refs:add_module_init_func (Results 1 – 2 of 2) sorted by relevance
339 trace_rules.add_module_init_func("einops", _allow_in_graph_einops)
3042 def add_module_init_func(name: str, init_func: Callable[[], None]) -> None: function