Searched refs:LocalBuffer (Results 1 – 2 of 2) sorted by relevance
402 class LocalBuffer(object): class421 return LocalBuffer(cbuf, backend, device)445 LocalBuffer(sub_buffer, device=self._device, backend=self._backend)823 return LocalBuffer(865 LocalBuffer(875 return LocalBuffer.from_pyval(885 return LocalBuffer.from_pyval(arg, device, backend=self._backend)
415 arg_buffers = [xla_client.LocalBuffer.from_pyval(arg) for arg in arguments]446 arg_buffer = xla_client.LocalBuffer.from_pyval(arg)453 local_buffer = xla_client.LocalBuffer.from_pyval(t)460 local_buffer = xla_client.LocalBuffer.from_pyval(t)472 local_buffer = xla_client.LocalBuffer.from_pyval(t)486 local_buffer = xla_client.LocalBuffer.from_pyval(t)502 local_buffer = xla_client.LocalBuffer.from_pyval(pyval)