Searched defs:parallel_device (Results 1 – 7 of 7) sorted by relevance
30 namespace parallel_device { namespace54 ParallelDevice parallel_device(std::move(devices)); in TEST() local110 ParallelDevice parallel_device(std::move(devices)); in TEST() local152 ParallelDevice parallel_device(devices); in TEST() local240 ParallelDevice parallel_device(std::move(devices)); in TEST() local305 ParallelDevice parallel_device(std::move(devices)); in TEST() local
31 namespace parallel_device { namespace54 std::unique_ptr<ParallelDevice> parallel_device) in NamedParallelDevice()65 const ParallelDevice& parallel_device, in ExecuteWithSpecialOps()378 std::unique_ptr<ParallelDevice> parallel_device( in AllocateParallelDevice() local
27 namespace parallel_device { namespace408 const ParallelDevice& parallel_device, in FromTensorHandles()427 const ParallelDevice& parallel_device, in FromTensorHandles()
45 namespace parallel_device { namespace
32 namespace parallel_device { namespace
33 namespace parallel_device { namespace
75 def __init__(self, parallel_device, wrapped_variable): argument104 def _variable_creator(next_creator, parallel_device, **kwargs): argument125 def independent_buffers(parallel_device): argument