Home
last modified time | relevance | path

Searched refs:devices_ (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dremote_device_test.cc40 std::vector<Device*> devices_; member in tensorflow::RemoteDeviceTest
71 devices_ = *found; in SetUp()
75 EXPECT_EQ(devices_.size(), 2); in SetUp()
76 std::sort(devices_.begin(), devices_.end(), [](Device* a, Device* b) { in SetUp()
82 for (auto d : devices_) delete d; in TearDown()
88 EXPECT_EQ(devices_[0]->name(), in TEST_F()
90 EXPECT_EQ(devices_[0]->attributes().device_type(), in TEST_F()
92 EXPECT_EQ(devices_[0]->attributes().memory_limit(), 256 << 20); in TEST_F()
93 EXPECT_EQ(devices_[1]->name(), in TEST_F()
95 EXPECT_EQ(devices_[1]->attributes().memory_limit(), 256 << 20); in TEST_F()
Dscheduler.h84 const DeviceSet* devices_; variable
109 const DeviceSet* devices_;
Dscheduler.cc192 : devices_(devices), in GreedyScheduler()
196 for (Device* d : devices_->devices()) { in GreedyScheduler()
296 : devices_(devices), cost_model_(cost_model), graph_(g) {} in PriorityScheduler()
303 GreedyScheduler greedysched(devices_, cost_model_, graph_, &slacks); in ComputeSchedule()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc28 std::unordered_map<string, DeviceProperties> devices_; member in tensorflow::grappler::__anone36516310111::GraphMemoryTest
32 devices_["/CPU:0"].set_type("CPU"); in GraphMemoryTest()
33 devices_["/CPU:0"].set_num_cores(1); in GraphMemoryTest()
34 devices_["/CPU:0"].set_frequency(1); in GraphMemoryTest()
35 devices_["/CPU:0"].set_bandwidth(1); in GraphMemoryTest()
37 devices_["/GPU:0"].set_type("GPU"); in GraphMemoryTest()
38 devices_["/GPU:0"].set_num_cores(1); in GraphMemoryTest()
39 devices_["/GPU:0"].set_frequency(1); in GraphMemoryTest()
40 devices_["/CPU:0"].set_bandwidth(1); in GraphMemoryTest()
41 (*devices_["/GPU:0"].mutable_environment())["architecture"] = "3"; in GraphMemoryTest()
[all …]
Dvirtual_placer.cc28 : devices_(devices), in VirtualPlacer()
32 lfqn_map_.reserve(devices_.size()); in VirtualPlacer()
33 for (const auto& kv : devices_) { in VirtualPlacer()
43 if (devices_.empty()) { in VirtualPlacer()
47 DeviceProperties& prop = devices_["UNKNOWN"]; in VirtualPlacer()
49 } else if (devices_.size() == 1) { in VirtualPlacer()
52 default_device_name_ = devices_.begin()->first; in VirtualPlacer()
85 default_device_name_ = devices_.begin()->first; // Any device. in VirtualPlacer()
119 auto it = devices_.find(device); in get_device()
120 DCHECK(it != devices_.end()); in get_device()
Dvirtual_placer.h50 std::unordered_map<string, DeviceProperties> devices_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.cc32 : devices_(std::move(devices)), in StaticDeviceMgr()
35 for (auto& d : devices_) { in StaticDeviceMgr()
65 for (auto& device : devices_) { in ~StaticDeviceMgr()
79 devices->reserve(devices_.size()); in ListDeviceAttributes()
80 for (const auto& dev : devices_) { in ListDeviceAttributes()
86 std::vector<Device*> devices(devices_.size()); in ListDevices()
87 for (size_t i = 0; i < devices_.size(); ++i) { in ListDevices()
88 devices[i] = devices_[i].get(); in ListDevices()
95 for (const auto& dev : devices_) { in DebugString()
103 for (const auto& dev : devices_) { in DeviceMappingString()
[all …]
Ddevice_set.cc35 devices_.push_back(device); in AddDevice()
49 for (Device* d : devices_) { in FindMatchingDevices()
80 for (Device* d : devices_) { in PrioritizedDeviceTypeList()
163 UpdatePrioritizedVectors(devices_, &prioritized_devices_, in prioritized_devices()
170 UpdatePrioritizedVectors(devices_, &prioritized_devices_, in prioritized_device_types()
Ddevice_mgr.h101 const std::vector<std::unique_ptr<Device>> devices_;
166 devices_.resize(kStaleDeviceBufferSize); in DeviceCircularBuffer()
169 devices_[index_] = std::move(device); in add()
175 std::vector<std::unique_ptr<Device>> devices_; variable
Ddevice_set_test.cc46 devices_.AddDevice(d); in AddDevice()
50 const DeviceSet& device_set() const { return devices_; } in device_set()
53 return devices_.PrioritizedDeviceTypeList(); in types()
57 DeviceSet devices_; member in tensorflow::__anon1b3a6df60111::DeviceSetTest
Dplacer.cc141 devices_(devices), in Placer()
159 if (devices_->devices().empty()) { in Run()
175 ColocationGraph colocation_graph(graph_, stack, flib_def_, devices_, in Run()
304 devices_->FindDeviceByName(candidate_device_name); in CanAssignToDevice()
Ddevice_set.h53 const std::vector<Device*>& devices() const { return devices_; } in devices()
110 std::vector<Device*> devices_; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable_run_options.h39 return H::combine(std::move(h), k.devices_); in AbslHashValue()
42 return a.devices_ == b.devices_;
45 const std::vector<GlobalDeviceId>& devices() const { return devices_; } in devices()
50 std::vector<GlobalDeviceId> devices_;
Dgpu_executable_run_options.cc24 : devices_(std::move(devices)) {} in NcclCliqueKey()
27 return GlobalDeviceIdsToString(devices_); in ToString()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item.cc64 item.devices_ = devices_; in WithGraph()
171 return devices_; in devices()
186 devices_.insert(DeviceNameUtils::ParsedNameToString(name)); in AddDevice()
209 VLOG(2) << "Inferred device set: [" << absl::StrJoin(devices_, ", ") << "]"; in InferDevicesFromGraph()
217 void GrapplerItem::ClearDevices() { devices_.clear(); } in ClearDevices()
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver.cc204 std::vector<PjRtDevice*> devices_; member in xla::OutfeedReceiverImpl
235 devices_.push_back(device); in OutfeedReceiverImpl()
238 CHECK_GT(devices_.size(), 0); in OutfeedReceiverImpl()
251 int num_threads = 1 + devices_.size(); in Start()
255 for (int device_idx = 0; device_idx < devices_.size(); ++device_idx) { in Start()
268 for (int device_idx = 0; device_idx < devices_.size(); ++device_idx) { in Shutdown()
287 PjRtDevice* device = devices_[device_idx]; in DeviceListenerThreadLoop()
388 const PjRtDevice* device = devices_[device_idx]; in SendShutdownOutfeedHeader()
408 devices_[device_idx]->client()->Compile( in SendShutdownOutfeedHeader()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_testlib.h64 const std::vector<DeviceAttributes>& devices() const { return devices_; } in devices()
71 std::vector<DeviceAttributes> devices_; variable
/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc119 for (auto& device : devices_) { in Run()
140 devices_.push_back(FakeDevice::Make( in AddDevices()
143 devices_.push_back(FakeDevice::Make( in AddDevices()
Doptimization_pass_runner.h53 std::vector<std::unique_ptr<Device>> devices_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc65 devices_(devices), in MultiDeviceIterator()
74 return strings::StrCat("MultiDeviceIterator for ", devices_.size(), in DebugString()
96 devices_.size(), max_buffer_size, incarnation_id_, std::move(iterator), in Init()
379 const std::vector<string> devices_; member in tensorflow::data::__anond56f2ce10111::MultiDeviceIterator
403 OP_REQUIRES_OK(ctx, ctx->GetAttr(kDevices, &devices_)); in MultiDeviceIteratorHandleOp()
445 context->env(), output_types_, output_shapes_, devices_, in Compute()
463 output_shapes_, devices_, in Compute()
505 std::vector<string> devices_; member in tensorflow::data::__anond56f2ce10111::MultiDeviceIteratorHandleOp
516 OP_REQUIRES_OK(ctx, ctx->GetAttr(kDevices, &devices_)); in AnonymousMultiDeviceIteratorOp()
532 devices_, std::move(flib_def), std::move(pflr), in CreateResource()
[all …]
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.h95 return devices_; in GetDevices()
137 std::unordered_map<string, DeviceProperties> devices_;
Dcluster.cc100 device_names.reserve(devices_.size()); in GetDeviceNames()
101 for (const auto& device : devices_) { in GetDeviceNames()
Dvirtual_cluster.cc37 devices_ = devices; in VirtualCluster()
54 devices_[device->name()] = props; in VirtualCluster()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node.h212 exec_.mutable_devices()->Reserve(devices_.size()); in ToProto()
213 for (const string& d : devices_) { in ToProto()
232 devices_.clear(); in FromProto()
233 devices_.insert(exec.devices().begin(), exec.devices().end()); in FromProto()
283 std::set<string> devices_; variable
789 devices_.clear(); in SnapshotNodes()
817 devices_.insert(node->canonical_device()); in SnapshotNodes()
852 const std::set<string>& devices() const { return devices_; } in devices()
900 std::set<string> devices_; variable
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc71 devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices()); in SetUpTestSuite()
72 VLOG(1) << "Running test with " << devices_->size() << " gpus"; in SetUpTestSuite()
73 if (devices_->size() <= 1) { in SetUpTestSuite()
80 ASSERT_GT(devices_->size(), 0) << "No GPUs found"; in SetUp()
84 static int32 NumGPUs() { return static_cast<int32>(devices_->size()); } in NumGPUs()
100 delete devices_; in TearDownTestSuite()
428 CHECK_LT(device_idx, devices_->size()); in GetDevice()
429 return (*devices_)[device_idx].get(); in GetDevice()
445 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_; member in tensorflow::NcclManagerTest
451 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ = member in tensorflow::NcclManagerTest<Scalar>

12