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()
34 std::unique_ptr<LocalDeviceState> local_device_state, in PjRtTpuDevice()
236 std::unique_ptr<LocalDeviceState> local_device_state; in GetTpuDevices() local
405 std::unique_ptr<LocalDeviceState> local_device_state, in GpuDevice()
102 LocalDeviceState* local_device_state() const { in local_device_state() function
1170 LocalDeviceState* local_device_state = device_->local_device_state(); in Release() local1587 LocalDeviceState* local_device_state = device_->local_device_state(); in GetReadyFuture() local