Searched refs:device_peak_memory (Results 1 – 4 of 4) sorted by relevance
569 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local570 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()571 EXPECT_EQ(device_peak_memory.size(), 1); in TEST_F()572 EXPECT_GT(device_peak_memory.begin()->second, 0); in TEST_F()595 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local596 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()598 device_peak_memory.find("/job:localhost/replica:0/task:0/device:CPU:0"), in TEST_F()599 device_peak_memory.end()); in TEST_F()601 device_peak_memory["/job:localhost/replica:0/task:0/device:CPU:0"]; in TEST_F()606 device_peak_memory.clear(); in TEST_F()[all …]
213 std::unordered_map<string, uint64>* device_peak_memory) const { in GetPeakMemoryUsage()225 device_peak_memory->clear(); in GetPeakMemoryUsage()233 (*device_peak_memory)[device->name()] = in GetPeakMemoryUsage()
116 std::unordered_map<string, uint64>* device_peak_memory) const { in GetPeakMemoryUsage() argument
52 std::unordered_map<string, uint64>* device_peak_memory) const override;