Home
last modified time | relevance | path

Searched refs:argnums (Results 1 – 25 of 25) sorted by relevance

/dports/math/py-autograd/autograd-1.3/autograd/
H A Dcore.py47 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 Dtracer.py43 argnums = tuple(argnum for argnum, _ in boxed_args)
45 node = node_constructor(ans, f_wrapped, argvals, kwargs, argnums, parents)
H A Dbuiltins.py119 list(g[key] for key in keys), argnums=(1,))
/dports/net-p2p/bazarr/bazarr-1.0.2/libs/future/types/
H A D__init__.py100 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__.py100 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__.py100 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__.py100 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 Dlinalg.py43 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 Dintegrate.py71 def build_selected_vjps(argnums, ans, combined_args, kwargs): argument
76 return [all_vjps[argnum] for argnum in argnums]
H A Dspecial.py17 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 Dpoisson.py15 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 Dchi2.py16 …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 Dt.py40 unbroadcast_f(loc, lambda g: -g * pdf(x, df, loc, scale)), argnums=(0,2))
57 argnums=(0,2))
H A Dgamma.py19 …a ans, x, a: unbroadcast_f(x, lambda g: g * np.exp(-x) * np.power(x, a-1) / gamma(a)), argnums=[0])
H A Dbeta.py22 …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 Dapi.py742 @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 Ddynamic_scale.py86 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 Djax2tf.py1065 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 Ddoubledouble.py266 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 Dnumpy_jvps.py79 defjvp(anp.full, 'same', argnums=(1,))
H A Dnumpy_vjps.py117 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 DChangeLog.03692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
/dports/devel/gettext-po-mode/gettext-0.21/gettext-tools/src/
H A DChangeLog.03692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
/dports/devel/gettext-runtime/gettext-0.21/gettext-tools/src/
H A DChangeLog.03692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.
/dports/devel/libtextstyle/gettext-0.21/gettext-tools/src/
H A DChangeLog.03692 (xgettext_lex, xgettext_lex_keyword): Add support for keyword argnums.