Searched refs:LazyExpr (Results 1 – 4 of 4) sorted by relevance
/dports/math/py-jax/jax-0.2.9/jax/ |
H A D | lazy.py | 92 LazyExpr = namedtuple('LazyExpr', ['input', 'shape', 'dims']) variable 101 return LazyExpr(ArrayVar(), shape, tuple(range(len(shape)))) 104 return LazyExpr(Iota(dtype, size), (size,), (0,)) 108 return LazyExpr(Eye(dtype, shape, offset), shape, (0, 1)) 112 return LazyExpr(Tri(dtype, shape, offset), shape, (0, 1)) 121 return LazyExpr(lexpr.input, shape, tuple(new_dims)) 123 def transpose(lexpr: LazyExpr, perm: Sequence[int]): argument 126 return LazyExpr(lexpr.input, new_shape, new_dims) 128 def is_constant(lexpr: Optional[LazyExpr]): argument 131 def is_trivial(lexpr: Optional[LazyExpr]) -> bool: argument [all …]
|
/dports/devel/py-zope.tales/zope.tales-3.5.2/src/zope/tales/ |
H A D | engine.py | 25 from zope.tales.expressions import LazyExpr 38 reg('lazy', LazyExpr)
|
H A D | expressions.py | 327 class LazyExpr(DeferExpr): class
|
/dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
H A D | xla.py | 1031 lazy_expr: Optional[lazy.LazyExpr], argument 1072 lazy_expr: Optional[lazy.LazyExpr], argument 1350 device: Device, lexpr: lazy.LazyExpr argument
|