Searched refs:device_peak_memory_before (Results 1 – 1 of 1) sorted by relevance
559 std::unordered_map<string, uint64> device_peak_memory_before; in TEST_F() local560 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_before)); in TEST_F()561 EXPECT_EQ(device_peak_memory_before.size(), 1); in TEST_F()563 EXPECT_LT(device_peak_memory_before.begin()->second, 400); in TEST_F()582 EXPECT_EQ(device_peak_memory_before.size(), 1); in TEST_F()584 EXPECT_LT(device_peak_memory_before.begin()->second, 400); in TEST_F()