Searched defs:local_device_state (Results 1 – 7 of 7) sorted by relevance
28 int id, std::unique_ptr<LocalDeviceState> local_device_state) in InterpreterDevice()
28 std::unique_ptr<LocalDeviceState> local_device_state) in CpuDevice()
32 std::unique_ptr<LocalDeviceState> local_device_state, in PjRtTpuDevice()
161 std::unique_ptr<LocalDeviceState> local_device_state; in GetTpuDevices() local
308 std::unique_ptr<LocalDeviceState> local_device_state, in GpuDevice()
95 LocalDeviceState* local_device_state() const { in local_device_state() function
1062 LocalDeviceState* local_device_state = in Release() local1448 LocalDeviceState* local_device_state = in BlockHostUntilReady() local