Searched refs:canonicalize_types (Results 1 – 4 of 4) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/lib/ |
| H A D | xla_bridge.py | 304 def _numpy_array_constant(builder, value, canonicalize_types=True): argument
305 if canonicalize_types:
322 def constant(builder, py_val, canonicalize_types=True): argument
333 return _constant_handlers[py_type](builder, py_val, canonicalize_types)
409 def _ndarray_constant_handler(c, val, canonicalize_types=True): argument
436 _numpy_array_constant(c, collapsed_val, canonicalize_types),
441 return _numpy_array_constant(c, val, canonicalize_types)
445 def _scalar_constant_handler(c, val, canonicalize_types=True): argument
446 return _numpy_array_constant(c, val, canonicalize_types)
|
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | pxla.py | 627 def _sharded_device_array_constant_handler(c, val, canonicalize_types=True): argument
628 return xb.constant(c, np.asarray(val), canonicalize_types=canonicalize_types)
|
| H A D | xla.py | 1295 def _device_array_constant_handler(c, val, canonicalize_types=True): argument
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | lax.py | 5523 canonicalize_types=False)
|