Searched refs:DeviceConstant (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 1040 not isinstance(device_buffer, DeviceConstant)):
1280 class DeviceConstant(object): class
1287 return type_is_device_array(x) and type(x.device_buffer) is DeviceConstant
1317 DeviceConstant(device))
|
| /dports/math/py-jax/jax-0.2.9/jax/_src/lax/ |
| H A D | lax.py | 1480 return xla._DeviceArray(aval, None, lazy_expr, xla.DeviceConstant())
1502 return xla._DeviceArray(aval, None, lazy_expr, xla.DeviceConstant())
1519 return xla._DeviceArray(aval, None, lazy_expr, xla.DeviceConstant())
1533 return xla._DeviceArray(aval, None, lazy_expr, xla.DeviceConstant())
|