Searched refs:device_peak_memory (Results 1 – 4 of 4) sorted by relevance
555 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local556 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()557 EXPECT_EQ(device_peak_memory.size(), 1); in TEST_F()558 EXPECT_GT(device_peak_memory.begin()->second, 0); in TEST_F()581 std::unordered_map<string, uint64> device_peak_memory; in TEST_F() local582 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()584 device_peak_memory.find("/job:localhost/replica:0/task:0/device:CPU:0"), in TEST_F()585 device_peak_memory.end()); in TEST_F()587 device_peak_memory["/job:localhost/replica:0/task:0/device:CPU:0"]; in TEST_F()592 device_peak_memory.clear(); in TEST_F()[all …]
216 std::unordered_map<string, uint64>* device_peak_memory) const { in GetPeakMemoryUsage()228 device_peak_memory->clear(); in GetPeakMemoryUsage()236 (*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;