Searched refs:device_array (Results 1 – 4 of 4) sorted by relevance
126 struct wl_array *device_array in feedback_handle_main_device() argument136 assert(device_array->size == sizeof(dev_id)); in feedback_handle_main_device()137 memcpy(&dev_id, device_array->data, sizeof(dev_id)); in feedback_handle_main_device()173 struct wl_array *device_array in feedback_handle_tranche_target_device() argument
269 py::object device_array = in DevicePut() local271 if (!device_array.is_none()) { in DevicePut()272 (*p)[device_array.ptr()] = HandleDeviceArray; in DevicePut()430 py::object device_array = in PyArgSignatureOfValue() local432 if (!device_array.is_none()) { in PyArgSignatureOfValue()433 (*p)[device_array.ptr()] = device_array_handler; in PyArgSignatureOfValue()
638 py::object device_array; in GetJitArgumentStickyDevice() member642 py::object device_array; in GetJitArgumentStickyDevice() local644 device_array = xla_module.attr("_DeviceArray"); in GetJitArgumentStickyDevice()646 return new PythonTypes{device_array}; in GetJitArgumentStickyDevice()659 if (arg.get_type().ptr() == types.device_array.ptr()) { in GetJitArgumentStickyDevice()
355 device_array = np.asarray([367 np.ravel(device_array).tolist(),