Searched refs:local_hardware_id (Results 1 – 5 of 5) sorted by relevance
84 virtual int local_hardware_id() const = 0;168 int local_hardware_id) const = 0;
240 int idx = device->local_hardware_id(); in PjRtStreamExecutorClient()964 /*allocator=*/nullptr, device->local_hardware_id(), in CreateViewOfDeviceBuffer()987 int local_hardware_id) const { in LookupAddressableDevice()989 if (local_hardware_id == device->local_hardware_id()) { in LookupAddressableDevice()994 local_hardware_id); in LookupAddressableDevice()2177 addressable_devices.front()->local_hardware_id()); in Compile()
90 int local_hardware_id() const override { return device_ordinal_; } in local_hardware_id() function154 int local_hardware_id) const override;
221 context.device_id = device.local_hardware_id(); in DLContextForDevice()289 dt.ctx.device_id = buffer->buffer()->device()->local_hardware_id(); in BufferToDLPackManagedTensor()
64 int local_hardware_id() const override { return -1; } in local_hardware_id() function