Searched defs:parallel_device (Results 1 – 7 of 7) sorted by relevance
30 namespace parallel_device { namespace55 ParallelDevice parallel_device(std::move(devices)); in TEST() local111 ParallelDevice parallel_device(std::move(devices)); in TEST() local153 ParallelDevice parallel_device(devices); in TEST() local241 ParallelDevice parallel_device(std::move(devices)); in TEST() local305 ParallelDevice parallel_device(std::move(devices)); in TEST() local
33 namespace parallel_device { namespace56 std::unique_ptr<ParallelDevice> parallel_device) in NamedParallelDevice()67 const ParallelDevice& parallel_device, in ExecuteWithSpecialOps()421 std::unique_ptr<ParallelDevice> parallel_device( in AllocateParallelDevice() local
32 namespace parallel_device { namespace505 const ParallelDevice& parallel_device, in FromTensorHandles()533 const ParallelDevice& parallel_device, in FromTensorHandles()
45 namespace parallel_device { namespace
32 namespace parallel_device { namespace
37 namespace parallel_device { namespace
162 const parallel_device::ParallelDevice& parallel_device() const { in parallel_device() function