| D | autograd_function_db.py | 47 def vmap(info, in_dims, input): member in NumpyCube
127 def vmap(info, in_dims, x, y): member in NumpyMul
198 def vmap(info, in_dims, x): member in NumpyExp_
238 def vmap(info, in_dims, x, dim): member in NumpySort
316 def vmap(info, in_dims, x, ind, ind_inv, dim): member in NumpyTake
384 def vmap(info, in_dims, x, idx): member in Select
495 def vmap(info, in_dims, x, idx): member in ForwardHasDefaultArgs
|