/dports/math/py-autograd/autograd-1.3/autograd/ |
H A D | core.py | 47 def vjp_argnums(argnums, *args): argument 53 argnums = kwargs.get('argnums', count()) 56 def vjp_argnums(argnums, ans, args, kwargs): argument 57 L = len(argnums) 60 argnum = argnums[0] 69 argnum_0, argnum_1 = argnums 131 argnums = kwargs.get('argnums', count()) 299 defvjp(primitive_fun, *vjpmakers, argnums=argnums) 305 def defvjp_is_zero(argnums=(0,)): argument 307 zero_vjps[0] |= set(argnums) [all …]
|
H A D | tracer.py | 43 argnums = tuple(argnum for argnum, _ in boxed_args) 45 node = node_constructor(ans, f_wrapped, argvals, kwargs, argnums, parents)
|
H A D | builtins.py | 119 list(g[key] for key in keys), argnums=(1,))
|
/dports/net-p2p/bazarr/bazarr-1.0.2/libs/future/types/ |
H A D | __init__.py | 100 def disallow_types(argnums, disallowed_types): argument 150 for (argnum, mytype) in zip(argnums, disallowed_types): 170 def no(mytype, argnums=(1,)): argument 190 if isinstance(argnums, Integral): 191 argnums = (argnums,) 192 disallowed_types = [mytype] * len(argnums) 193 return disallow_types(argnums, disallowed_types)
|
/dports/multimedia/tautulli/Tautulli-2.8.0/lib/future/types/ |
H A D | __init__.py | 100 def disallow_types(argnums, disallowed_types): argument 150 for (argnum, mytype) in zip(argnums, disallowed_types): 170 def no(mytype, argnums=(1,)): argument 190 if isinstance(argnums, Integral): 191 argnums = (argnums,) 192 disallowed_types = [mytype] * len(argnums) 193 return disallow_types(argnums, disallowed_types)
|
/dports/www/chromium-legacy/chromium-88.0.4324.182/third_party/catapult/third_party/google-endpoints/future/types/ |
H A D | __init__.py | 100 def disallow_types(argnums, disallowed_types): argument 150 for (argnum, mytype) in zip(argnums, disallowed_types): 170 def no(mytype, argnums=(1,)): argument 190 if isinstance(argnums, Integral): 191 argnums = (argnums,) 192 disallowed_types = [mytype] * len(argnums) 193 return disallow_types(argnums, disallowed_types)
|
/dports/devel/py-future/future-0.18.2/src/future/types/ |
H A D | __init__.py | 100 def disallow_types(argnums, disallowed_types): argument 150 for (argnum, mytype) in zip(argnums, disallowed_types): 170 def no(mytype, argnums=(1,)): argument 190 if isinstance(argnums, Integral): 191 argnums = (argnums,) 192 disallowed_types = [mytype] * len(argnums) 193 return disallow_types(argnums, disallowed_types)
|
/dports/math/py-autograd/autograd-1.3/autograd/scipy/ |
H A D | linalg.py | 43 def _jvp_sylvester(argnums, dms, ans, args, _): argument 45 if 0 in argnums: 47 db = dms[1] if 1 in argnums else 0 50 db = dms[0] if 1 in argnums else 0 51 dq = dms[-1] if 2 in argnums else 0 56 def _vjp_sylvester(argnums, ans, args, _): argument 61 if 0 in argnums: vjps.append(-anp.dot(q_vjp, anp.transpose(ans))) 62 if 1 in argnums: vjps.append(-anp.dot(anp.transpose(ans), q_vjp)) 63 if 2 in argnums: vjps.append(q_vjp)
|
H A D | integrate.py | 71 def build_selected_vjps(argnums, ans, combined_args, kwargs): argument 76 return [all_vjps[argnum] for argnum in argnums]
|
H A D | special.py | 17 argnums=[2]) 50 defvjp(gammainc, make_gammainc_vjp_arg1(1), argnums=[1]) 51 defvjp(gammaincc, make_gammainc_vjp_arg1(-1), argnums=[1])
|
/dports/math/py-autograd/autograd-1.3/autograd/scipy/stats/ |
H A D | poisson.py | 15 defvjp(cdf, lambda ans, k, mu: unbroadcast_f(mu, lambda g: g * -pmf(np.floor(k), mu)), argnums=[1]) 16 …ogpmf, lambda ans, k, mu: unbroadcast_f(mu, lambda g: g * grad_poisson_logpmf(k, mu)), argnums=[1]) 17 … lambda ans, k, mu: unbroadcast_f(mu, lambda g: g * ans * grad_poisson_logpmf(k, mu)), argnums=[1])
|
H A D | chi2.py | 16 …mbda g: g * np.power(2., -df/2) * np.exp(-x/2) * np.power(x, df/2 - 1) / gamma(df/2)), argnums=[0]) 17 defvjp(logpdf, lambda ans, x, df: unbroadcast_f(x, lambda g: g * grad_chi2_logpdf(x, df)), argnums=… 18 …pdf, lambda ans, x, df: unbroadcast_f(x, lambda g: g * ans * grad_chi2_logpdf(x, df)), argnums=[0])
|
H A D | t.py | 40 unbroadcast_f(loc, lambda g: -g * pdf(x, df, loc, scale)), argnums=(0,2)) 57 argnums=(0,2))
|
H A D | gamma.py | 19 …a ans, x, a: unbroadcast_f(x, lambda g: g * np.exp(-x) * np.power(x, a-1) / gamma(a)), argnums=[0])
|
H A D | beta.py | 22 …b: unbroadcast_f(x, lambda g: g * np.power(x, a-1) * np.power(1-x, b-1) / beta(a, b)), argnums=[0])
|
/dports/math/py-jax/jax-0.2.9/jax/ |
H A D | api.py | 742 @wraps(fun, docstr=docstr, argnums=argnums) 748 @wraps(fun, docstr=docstr, argnums=argnums) 793 argnums = _ensure_index(argnums) 795 @wraps(fun, docstr=docstr, argnums=argnums) 798 max_argnum = argnums if isinstance(argnums, int) else max(argnums) 805 f_partial, dyn_args = argnums_partial(f, argnums, args) 815 g = g[0] if isinstance(argnums, int) else g 895 argnums = _ensure_index(argnums) 899 f_partial, dyn_args = argnums_partial(f, argnums, args) 972 jac = jac[0] if isinstance(argnums, int) else jac [all …]
|
/dports/math/py-flax/flax-0.3.3/flax/optim/ |
H A D | dynamic_scale.py | 86 argnums: Union[int, Sequence[int]] = 0, 117 grad_fn = jax.value_and_grad(loss_wrapper, argnums, has_aux)
|
/dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/ |
H A D | jax2tf.py | 1065 def bool_to_int8(f, argnums): argument 1067 argnums = tf.nest.flatten(argnums) 1069 if not any(args[i].dtype == tf.bool for i in argnums): 1072 args_cast = [(tf.cast(a, tf.int8) if i in argnums else a) 1077 _in_avals_cast = [cast_aval(aval) if i in argnums else aval 1438 bool_to_int8(axes_to_axis(tf.reduce_sum), argnums=0)) 1440 bool_to_int8(axes_to_axis(tf.reduce_prod), argnums=0)) 1442 bool_to_int8(axes_to_axis(tf.reduce_max), argnums=0)) 1444 bool_to_int8(axes_to_axis(tf.reduce_min), argnums=0)) 1765 @functools.partial(bool_to_int8, argnums=(0, 1)) [all …]
|
/dports/math/py-jax/jax-0.2.9/jax/experimental/ |
H A D | doubledouble.py | 266 def _def_passthrough(prim, argnums=(0,)): argument 269 prim.bind(*(arg[0] if i in argnums else arg for i, arg in enumerate(args)), **kwargs), 270 prim.bind(*(arg[1] if i in argnums else arg for i, arg in enumerate(args)), **kwargs)
|
/dports/math/py-autograd/autograd-1.3/autograd/numpy/ |
H A D | numpy_jvps.py | 79 defjvp(anp.full, 'same', argnums=(1,))
|
H A D | numpy_vjps.py | 117 defvjp(anp.full, lambda ans, shape, fill_value, dtype=None : lambda g: anp.sum(g), argnums=(1,)) 691 defvjp(anp._array_from_scalar_or_array, array_from_scalar_or_array_gradmaker, argnums=(2,3))
|
/dports/devel/gettext-tools/gettext-0.21/gettext-tools/src/ |
H A D | ChangeLog.0 | 3692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
|
/dports/devel/gettext-po-mode/gettext-0.21/gettext-tools/src/ |
H A D | ChangeLog.0 | 3692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
|
/dports/devel/gettext-runtime/gettext-0.21/gettext-tools/src/ |
H A D | ChangeLog.0 | 3692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
|
/dports/devel/libtextstyle/gettext-0.21/gettext-tools/src/ |
H A D | ChangeLog.0 | 3692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
|