Searched refs:vmap_function (Results 1 – 1 of 1) sorted by relevance
321 interpreter, vmap_function, op, *operands, **kwargs argument340 result = vmap_function(info, in_dims, *unwrapped_operands, **kwargs)