Searched refs:_scalar_types (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 77 _scalar_types = dtypes.python_scalar_dtypes.keys() variable
147 device_put_handlers.update((t, _device_put_scalar) for t in _scalar_types)
170 (t, partial(_canonicalize_python_scalar_dtype, t)) for t in _scalar_types)
189 (t, partial(_make_abstract_python_scalar, t)) for t in _scalar_types)
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/numpy/ |
| H A D | lax_numpy.py | 5101 _scalar_types = (int, float, complex, np.generic) variable
5106 if not isinstance(other, _scalar_types + _arraylike_types + (core.Tracer,)):
|