Searched refs:GetPeakMemoryUsage (Results 1 – 13 of 13) 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()
360 const std::unordered_map<string, int64> GetPeakMemoryUsage() const;499 const std::unordered_map<string, int64> GetPeakMemoryUsage() const { in GetPeakMemoryUsage() function500 return scheduler_state_->GetPeakMemoryUsage(); in GetPeakMemoryUsage()
55 const MemoryUsage& GetPeakMemoryUsage(const string& device) const { in GetPeakMemoryUsage() function
2536 scheduler_->GetPeakMemoryUsage()); in TEST_F()2610 scheduler_->GetPeakMemoryUsage()); in TEST_F()2699 scheduler_->GetPeakMemoryUsage()); in TEST_F()
1277 const std::unordered_map<string, int64> SchedulerState::GetPeakMemoryUsage() in GetPeakMemoryUsage() function in tensorflow::grappler::SchedulerState
560 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_before)); in TEST_F()570 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()578 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory_after)); in TEST_F()596 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()607 TF_CHECK_OK(cluster_->GetPeakMemoryUsage(&device_peak_memory)); in TEST_F()629 Status s = cluster.GetPeakMemoryUsage(&device_peak_memory); in TEST_F()
115 virtual Status GetPeakMemoryUsage( in GetPeakMemoryUsage() function
51 Status GetPeakMemoryUsage(
93 estimator_->GetScheduler()->GetPeakMemoryUsage(); in Run()
212 Status SingleMachine::GetPeakMemoryUsage( in GetPeakMemoryUsage() function in tensorflow::grappler::SingleMachine
101 auto status = cluster->GetPeakMemoryUsage(&peak_mem); in PrintDebugInfo()
320 memory.GetPeakMemoryUsage(device.first); in PYBIND11_MODULE()
561 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in SchedulingPass()1007 const GraphMemory::MemoryUsage& mem_usage = memory.GetPeakMemoryUsage(name); in IdentifySwappingCandidates()