Home
last modified time | relevance | path

Searched refs:abstractify (Results 1 – 9 of 9) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/experimental/
H A Dloops.py265 new_aval = flat_tree.unflatten(safe_map(_BodyTracer.abstractify, flat_values))
380 flat_init_avals = safe_map(_BodyTracer.abstractify, flat_init_vals)
390 index_var_aval = _BodyTracer.abstractify(0)
414 new_aval = flat_new_tree.unflatten(safe_map(_BodyTracer.abstractify, flat_new_values))
453 def abstractify(x): member in _BodyTracer
539 in_avals = safe_map(_BodyTracer.abstractify, in_vals)
581 init_avals = safe_map(_BodyTracer.abstractify, init_vals)
/dports/math/py-jax/jax-0.2.9/jax/
H A Dapi.py612 def abstractify(x): function
640 avals = map(abstractify, args_flat)
1971 def abstractify(x): function
1975 in_avals = map(abstractify, primals_flat)
2259 xla.abstractify(arg) # faster than core.get_aval
2355 def abstractify(x): function
2360 *map(abstractify, args_flat))
H A Dcustom_derivatives.py904 in_avals = tuple(map(abstractify, flat_args))
950 def abstractify(x): function
/dports/math/py-flax/flax-0.3.3/flax/
H A Djax_utils.py161 aval = jax.xla.abstractify(xs)
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py172 def abstractify(x) -> core.AbstractValue: function
230 aval = abstractify(x)
419 return abstractify(v.val)
1386 a = abstractify(x)
H A Dsharded_jit.py250 *map(xla.abstractify, args))
H A Dpxla.py642 abstract_args = unsafe_map(xla.abstractify, args)
1144 aval = xla.abstractify(val) # type: ShapedArray
1604 abstract_args = unsafe_map(xla.abstractify, args)
/dports/math/py-jax/jax-0.2.9/jax/experimental/jax2tf/
H A Djax2tf.py381 raw_aval = xla.abstractify(val)
/dports/editors/texstudio/texstudio-4.1.2/utilities/dictionaries/
H A Den_GB.dic373 abstractify/SGD