| /dports/math/py-jax/jax-0.2.9/jax/ |
| H A D | custom_derivatives.py | 490 out_trees=out_trees)
496 *args_flat, out_trees=out_trees)
500 *args_flat, out_trees=out_trees)
538 out_tree, res_tree = out_trees()
569 def bind(self, fun, fwd, bwd, *args, out_trees): argument
579 out_trees=out_trees)
584 del fwd, bwd, out_trees
612 out_tree, res_tree = out_trees()
657 out_trees=out_trees, num_consts=num_consts)
712 out_trees=out_trees)
[all …]
|
| H A D | core.py | 443 def process_custom_vjp_call(self, primitive, fun, fwd, bwd, tracers, out_trees): argument
638 def process_custom_vjp_call(self, primitive, fun, fwd, bwd, tracers, out_trees): argument
639 del primitive, fwd, bwd, out_trees # Unused.
|
| /dports/audio/sphinx3/sphinx3-0.8/src/libs3decoder/libam/ |
| H A D | kdtree.c | 185 read_kd_trees(const char *infile, kd_tree_t *** out_trees, in read_kd_trees() argument
210 *out_trees = ckd_calloc(*out_n_trees, sizeof(kd_tree_t **)); in read_kd_trees()
224 (*out_trees)[i] = tree = ckd_calloc(1, sizeof(*tree)); in read_kd_trees()
254 free_kd_tree((*out_trees)[i]); in read_kd_trees()
255 (*out_trees)[i] = NULL; in read_kd_trees()
257 ckd_free(*out_trees); in read_kd_trees()
258 *out_trees = NULL; in read_kd_trees()
|
| /dports/biology/iqtree/IQ-TREE-2.0.6/terraphast/lib/ |
| H A D | subtree_extraction.cpp | 109 vector<tree> out_trees; in subtrees() local
112 out_trees.push_back(subtree(t, node_occ, num_leaves_per_site, site)); in subtrees()
115 return out_trees; in subtrees()
|
| /dports/audio/sphinx3/sphinx3-0.8/include/ |
| H A D | kdtree.h | 75 int32 read_kd_trees(const char *infile, kd_tree_t ***out_trees, uint32 *out_n_trees,
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/ |
| H A D | callback.py | 167 out_trees): argument
170 del primitive, fwd, bwd, out_trees # Unused.
|
| H A D | jet.py | 161 def process_custom_vjp_call(self, primitive, fun, fwd, bwd, tracers, out_trees): argument
162 del primitive, fwd, bwd, out_trees # Unused.
|
| H A D | host_callback.py | 1293 out_trees="illegal param"
|
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | partial_eval.py | 345 def process_custom_vjp_call(self, prim, fun, fwd, bwd, tracers, out_trees): argument
350 out_flat = prim.bind(fun, fwd, bwd, *in_consts, out_trees=out_trees)
374 bwd=bwd, out_trees=out_trees),
1144 def process_custom_vjp_call(self, prim, fun, fwd, bwd, tracers, out_trees): argument
1158 bwd=bwd, out_trees=out_trees),
1349 def process_custom_vjp_call(self, prim, fun, fwd, bwd, tracers, out_trees): argument
|
| H A D | batching.py | 251 def process_custom_vjp_call(self, prim, fun, fwd, bwd, tracers, *, out_trees): argument
259 out_vals = prim.bind(fun, fwd, bwd, *in_vals, out_trees=out_trees)
|
| H A D | ad.py | 359 def process_custom_vjp_call(self, _, __, fwd, bwd, tracers, *, out_trees): argument
363 out_tree, res_tree = out_trees()
|
| /dports/misc/py-xgboost/xgboost-1.5.1/src/gbm/ |
| H A D | gbtree.cc | 453 std::vector<std::unique_ptr<RegTree>> &out_trees = out_model.trees; in Slice() local
454 out_trees.resize(layer_trees * n_layers); in Slice()
472 out_trees.at(out_it) = std::move(new_tree); in Slice()
|
| /dports/misc/xgboost/xgboost-1.5.1/src/gbm/ |
| H A D | gbtree.cc | 453 std::vector<std::unique_ptr<RegTree>> &out_trees = out_model.trees; in Slice() local
454 out_trees.resize(layer_trees * n_layers); in Slice()
472 out_trees.at(out_it) = std::move(new_tree); in Slice()
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | control_flow.py | 602 jaxprs, consts, out_trees = _initial_style_jaxprs_with_common_consts(
605 for i, (out_tree, jaxpr) in enumerate(zip(out_trees[1:], jaxprs[1:])):
607 out_trees[0], jaxprs[0].out_avals,
613 return tree_unflatten(out_trees[0], out)
691 jaxprs, consts, out_trees = _initial_style_jaxprs_with_common_consts(
694 out_tree, false_out_tree = out_trees
|
| /dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/ |
| H A D | jax2tf.py | 784 def process_custom_vjp_call(self, prim, fun, fwd, bwd, tracers, out_trees): argument
788 del fwd, bwd, out_trees # Unused.
|