Searched refs:instances_ (Results 1 – 6 of 6) sorted by relevance
134 instances_.push_back(absl::make_unique<DeviceInstance>( in Init()143 for (const auto& instance : instances_) { in Reduce()154 while (done < instances_.size()) done_cv.wait(l); in Reduce()161 DeviceInstance* instance = instances_[rank].get(); in RunTest()176 for (int rank = 0; rank < instances_.size(); ++rank) { in RunTest()177 TF_ASSERT_OK(instances_[rank]->status_); in RunTest()178 Tensor* dev_tensor = &instances_[rank]->tensor_; in RunTest()181 Device* dev = instances_[rank]->device_; in RunTest()318 std::vector<std::unique_ptr<DeviceInstance>> instances_; member in tensorflow::NcclReducerTest
129 for (auto i : instances_) delete i; in ~RingGathererTest()237 instances_.push_back(new DeviceInstance( in Init()245 for (auto di : instances_) { in Gather()255 while (done < static_cast<int>(instances_.size())) { in Gather()268 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()269 DeviceInstance* instance = instances_[di]; in RunTest()290 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()292 instances_[di]->status_.error_message()); in RunTest()297 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()298 TF_EXPECT_OK(instances_[di]->status_); in RunTest()[all …]
197 for (auto i : instances_) delete i; in ~HierarchicalTreeBroadcasterTest()334 instances_.push_back(new DeviceInstance( in Init()344 VLOG(2) << "#instances=" << instances_.size(); in Broadcast()346 for (auto di : instances_) { in Broadcast()352 while (done < instances_.size()) { in Broadcast()413 for (int di = 0; di < instances_.size(); ++di) { in RunTest()414 DeviceInstance* instance = instances_[di]; in RunTest()428 const CollectiveParams& cp = instances_[0]->col_params_; in RunTest()432 const Tensor* t = &instances_[broadcast_dev_id]->tensor_; in RunTest()436 Device* dev = instances_[broadcast_dev_id]->device_; in RunTest()[all …]
151 for (auto i : instances_) delete i; in ~RingReducerTest()259 instances_.push_back(new DeviceInstance( in Init()267 for (auto di : instances_) { in Reduce()277 while (done < static_cast<int>(instances_.size())) { in Reduce()289 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()290 DeviceInstance* instance = instances_[di]; in RunTest()308 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()310 instances_[di]->status_.error_message()); in RunTest()317 for (int di = 0; di < static_cast<int>(instances_.size()); ++di) { in RunTest()318 TF_EXPECT_OK(instances_[di]->status_); in RunTest()[all …]
49 EventCollector::instances_; member in tensorflow::tracing::EventCollector53 EventCollector::instances_[static_cast<unsigned>(category)] = collector; in SetEventCollector()
66 static std::array<const EventCollector*, GetNumEventCategories()> instances_; variable76 return EventCollector::instances_[static_cast<unsigned>(category)]; in GetEventCollector()