Searched refs:MemoryUsage (Results 1 – 6 of 6) sorted by relevance
38 struct MemoryUsage { struct55 const MemoryUsage& GetPeakMemoryUsage(const string& device) const { in GetPeakMemoryUsage() argument74 std::unordered_map<string, MemoryUsage> peak_usage_;75 const MemoryUsage unknown_usage_;
54 const GraphMemory::MemoryUsage& mem_usage = in TEST_F()84 const GraphMemory::MemoryUsage& mem_usage = in TEST_F()110 const GraphMemory::MemoryUsage& cpu_mem = memory.GetPeakMemoryUsage("/CPU:0"); in TEST_F()123 const GraphMemory::MemoryUsage& gpu_mem = memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F()149 const GraphMemory::MemoryUsage& gpu_mem = in TEST_F()176 const GraphMemory::MemoryUsage& new_gpu_mem = in TEST_F()213 const GraphMemory::MemoryUsage& mem = memory.GetPeakMemoryUsage("/CPU:0"); in TEST_F()
283 MemoryUsage& peak_mem_usage = peak_usage_[live_per_device.first]; in InferFromTrace()
41 FRIEND_TEST(VirtualSchedulerTest, MemoryUsage);2389 TEST_F(VirtualSchedulerTest, MemoryUsage) { in TEST_F() argument
410 const tensorflow::grappler::GraphMemory::MemoryUsage& usage = in TF_DeterminePeakMemoryUsage()
552 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in SchedulingPass()972 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in IdentifySwappingCandidates()