Home
last modified time | relevance | path

Searched refs:instances_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer_test.cc134 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
/external/tensorflow/tensorflow/core/common_runtime/
Dring_gatherer_test.cc129 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 …]
Dhierarchical_tree_broadcaster_test.cc197 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 …]
Dring_reducer_test.cc151 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 …]
/external/tensorflow/tensorflow/core/platform/
Dtracing.cc49 EventCollector::instances_; member in tensorflow::tracing::EventCollector
53 EventCollector::instances_[static_cast<unsigned>(category)] = collector; in SetEventCollector()
Dtracing.h66 static std::array<const EventCollector*, GetNumEventCategories()> instances_; variable
76 return EventCollector::instances_[static_cast<unsigned>(category)]; in GetEventCollector()