Searched refs:DeviceThread (Results 1 – 2 of 2) sorted by relevance
61 class DeviceThread { class64 explicit DeviceThread(const std::string& device, const bool is_async) in DeviceThread() function in tensorflow::parallel_device::DeviceThread77 std::bind(&DeviceThread::Run, this))) {} in DeviceThread()78 ~DeviceThread();145 DeviceThread::~DeviceThread() { in ~DeviceThread()153 void DeviceThread::Run() { in Run()175 void DeviceThread::StartExecute(TFE_Context* context, in StartExecute()199 std::vector<TensorHandlePtr> DeviceThread::Join(TF_Status* status) { in Join()221 void DeviceThread::Execute(TFE_Context* context, const char* operation_name, in Execute()265 new DeviceThread(devices[device_index].c_str(), is_async)); in ParallelDevice()[all …]
49 class DeviceThread; variable149 std::vector<std::unique_ptr<DeviceThread>> device_threads_;