Searched refs:NamedParallelDevice (Results 1 – 1 of 1) sorted by relevance
51 class NamedParallelDevice { class53 NamedParallelDevice(const std::string& name, in NamedParallelDevice() function in tensorflow::parallel_device::__anon5d26a7fc0111::NamedParallelDevice239 NamedParallelDevice* named_device = in CopyToParallelDevice()240 reinterpret_cast<NamedParallelDevice*>(device_info); in CopyToParallelDevice()294 NamedParallelDevice* named_device = in ParallelDeviceExecute()295 reinterpret_cast<NamedParallelDevice*>(device_info); in ParallelDeviceExecute()359 delete reinterpret_cast<NamedParallelDevice*>(device_info); in DeleteParallelDevice()381 new NamedParallelDevice{device_name, std::move(parallel_device)}; in AllocateParallelDevice()