Searched refs:scatter_add_fn (Results 1 – 2 of 2) sorted by relevance
235 scatter_add_fn = lambda var, *a, **kw: var.scatter_add(*a, **kw) function237 update_fn=scatter_add_fn,
187 scatter_add_fn = lambda var, *a, **kw: var.scatter_add(*a, **kw) function189 update_fn=scatter_add_fn,