Home
last modified time | relevance | path

Searched refs:LazyExpr (Results 1 – 4 of 4) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/
H A Dlazy.py92 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 Dengine.py25 from zope.tales.expressions import LazyExpr
38 reg('lazy', LazyExpr)
H A Dexpressions.py327 class LazyExpr(DeferExpr): class
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py1031 lazy_expr: Optional[lazy.LazyExpr], argument
1072 lazy_expr: Optional[lazy.LazyExpr], argument
1350 device: Device, lexpr: lazy.LazyExpr argument