Searched refs:GetPeakMemoryUsage (Results 1 – 12 of 12) sorted by relevance
55 memory.GetPeakMemoryUsage("/CPU:0"); in TEST_F()85 memory.GetPeakMemoryUsage("/CPU:0"); 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()150 memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F()177 memory.GetPeakMemoryUsage("/GPU:0"); in TEST_F()213 const GraphMemory::MemoryUsage& mem = memory.GetPeakMemoryUsage("/CPU:0"); in TEST_F()
55 const MemoryUsage& GetPeakMemoryUsage(const string& device) const { in GetPeakMemoryUsage() function
292 const std::unordered_map<string, int64> GetPeakMemoryUsage() const;
1078 const std::unordered_map<string, int64> VirtualScheduler::GetPeakMemoryUsage() in GetPeakMemoryUsage() function in tensorflow::grappler::VirtualScheduler
546 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_before)); in TEST_F()556 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()564 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_after)); in TEST_F()582 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()593 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()615 Status s = cluster.GetPeakMemoryUsage(&device_peak_memory); in TEST_F()
115 virtual Status GetPeakMemoryUsage( in GetPeakMemoryUsage() function
51 Status GetPeakMemoryUsage(
91 estimator_->GetScheduler()->GetPeakMemoryUsage(); in Run()
215 Status SingleMachine::GetPeakMemoryUsage( in GetPeakMemoryUsage() function in tensorflow::grappler::SingleMachine
99 auto status = cluster->GetPeakMemoryUsage(&peak_mem); in PrintDebugInfo()
411 memory.GetPeakMemoryUsage(device.first); 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()