Searched refs:DeviceArrayBase (Results 1 – 4 of 4) sorted by relevance
30 from .xla_extension import DeviceArrayBase as DeviceArrayBase
394 DeviceArrayBase = _xla.DeviceArrayBase variable
650 if not isinstance(buffer, xla_client.DeviceArrayBase):
420 class DeviceArrayBase: ...422 class DeviceArray(DeviceArrayBase):