Searched refs:device_state (Results 1 – 8 of 8) sorted by relevance
1641 LocalDeviceState* device_state = &(client_->device_state(device_ordinal)); in MakeExecutionInputsAndWaitForEvents() local1647 MakeTupleHelper(client_, device_state, argument_handles, in MakeExecutionInputsAndWaitForEvents()1672 event->WaitForEventOnStream(device_state->compute_stream()); in MakeExecutionInputsAndWaitForEvents()1690 LocalDeviceState* device_state = &(client_->device_state(device_ordinal)); in EnqueueExecution() local1749 run_options.set_stream(device_state->compute_stream()); in EnqueueExecution()1750 run_options.set_host_to_device_stream(device_state->host_to_device_stream()); in EnqueueExecution()1756 run_options.set_rng_seed(device_state->GetNewPrngSeed()); in EnqueueExecution()1769 device_state->compute_semaphore().ScopedAcquire(1)); in EnqueueExecution()1782 if (device_state->allocation_model() == LocalDeviceState::kSynchronous) { in EnqueueExecution()1799 device_state->ThenExecuteOnCallbackThread( in EnqueueExecution()[all …]
47 auto device_state = absl::make_unique<LocalDeviceState>( in GetInterpreterClient() local51 absl::make_unique<InterpreterDevice>(0, std::move(device_state)); in GetInterpreterClient()
53 auto device_state = absl::make_unique<LocalDeviceState>( in GetCpuClient() local56 auto device = absl::make_unique<CpuDevice>(i, std::move(device_state)); in GetCpuClient()
214 LocalDeviceState& device_state(int device_ordinal) const { in device_state() function
69 std::string device_state; in is_locked_and_not_forced() local71 device_state = android::base::GetProperty("ro.boot.vbmeta.device_state", ""); in is_locked_and_not_forced()72 if (device_state == "locked" && !g_opt_force) { in is_locked_and_not_forced()
92 auto device_state = absl::make_unique<LocalDeviceState>( in GetCpuClientWithNonLocalDevice() local97 devices.push_back(absl::make_unique<CpuDevice>(0, std::move(device_state))); in GetCpuClientWithNonLocalDevice()
603 __u32 device_state; /* VFIO device state */ member
562 auto& device_state = device_[curr_node_device]; in Init() local566 device_state.persistent_nodes.insert( in Init()