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 { namespace468 const ParallelDevice& parallel_device, in FromTensorHandles()487 const ParallelDevice& parallel_device, in FromTensorHandles()
45 namespace parallel_device { namespace
32 namespace parallel_device { namespace
37 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