Searched defs:PyLocalBuffer (Results 1 – 2 of 2) sorted by relevance
54 PyLocalBuffer = Any variable1032 device_buffer: Union[PyLocalBuffer, "DeviceConstant"],1073 device_buffer: PyLocalBuffer):
507 device_buffers: List[xb.xla_client._xla.PyLocalBuffer] = None,