Searched refs:udf_module (Results 1 – 1 of 1) sorted by relevance
2178 par_mod = functools.partial(udf_module, mod=SmallNN())2183 par_mod = functools.partial(udf_module, mod=SmallNN().forward)3347 def udf_module(mod, x, y): function