Searched refs:abstractify (Results 1 – 9 of 9) sorted by relevance
/dports/math/py-jax/jax-0.2.9/jax/experimental/ |
H A D | loops.py | 265 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 D | api.py | 612 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 D | custom_derivatives.py | 904 in_avals = tuple(map(abstractify, flat_args)) 950 def abstractify(x): function
|
/dports/math/py-flax/flax-0.3.3/flax/ |
H A D | jax_utils.py | 161 aval = jax.xla.abstractify(xs)
|
/dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
H A D | xla.py | 172 def abstractify(x) -> core.AbstractValue: function 230 aval = abstractify(x) 419 return abstractify(v.val) 1386 a = abstractify(x)
|
H A D | sharded_jit.py | 250 *map(xla.abstractify, args))
|
H A D | pxla.py | 642 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 D | jax2tf.py | 381 raw_aval = xla.abstractify(val)
|
/dports/editors/texstudio/texstudio-4.1.2/utilities/dictionaries/ |
H A D | en_GB.dic | 373 abstractify/SGD
|