Searched refs:device_thread (Results 1 – 2 of 2) sorted by relevance
326 DeviceThread* device_thread = device_threads_[device_index].get(); in StartExecute() local333 device_thread->StartExecute(context, operation_name, in StartExecute()351 DeviceThread* device_thread = device_threads_[device_index].get(); in Join() local352 per_device_output_tensors.push_back(device_thread->Join(status)); in Join()
548 for (const auto& device_thread : thread_names_) { in FinalizeInternal() local549 if (dev_stats_pb.find(device_thread.first) == dev_stats_pb.end()) { in FinalizeInternal()553 DeviceStepStats* dss = dev_stats_pb.at(device_thread.first); in FinalizeInternal()554 for (const auto& thread_name : device_thread.second) { in FinalizeInternal()