Searched refs:parallel_device_ (Results 1 – 3 of 3) sorted by relevance
57 : device_name_(name), parallel_device_(std::move(parallel_device)) {} in NamedParallelDevice()59 const ParallelDevice& device() const { return *parallel_device_; } in device()63 std::unique_ptr<ParallelDevice> parallel_device_; member in tensorflow::parallel_device::__anon5736cf2b0111::NamedParallelDevice
149 parallel_device_(std::move(parallel_device)), in mesh_config_()163 return *parallel_device_; in parallel_device()189 std::unique_ptr<parallel_device::ParallelDevice> parallel_device_; variable
485 parallel_device_->ScalarsFromSequence<int32_t>(ids, context, status); in DeviceIDs()