Searched refs:numpy_view_copy_vmap (Results 1 – 1 of 1) sorted by relevance
248 def numpy_view_copy_vmap(info, in_dims, x, shape): function256 numpy_view_copy.register_vmap(numpy_view_copy_vmap)