Home
last modified time | relevance | path

Searched defs:GetPeakMemoryUsage (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dcluster.h115 virtual Status GetPeakMemoryUsage( in GetPeakMemoryUsage() function
H A Dsingle_machine.cc214 Status SingleMachine::GetPeakMemoryUsage( in GetPeakMemoryUsage() function in tensorflow::grappler::SingleMachine
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_memory.h55 const MemoryUsage& GetPeakMemoryUsage(const string& device) const { in GetPeakMemoryUsage() function
H A Dvirtual_scheduler.h514 const std::unordered_map<string, int64_t> GetPeakMemoryUsage() const { in GetPeakMemoryUsage() function
H A Dvirtual_scheduler.cc1333 const std::unordered_map<string, int64_t> SchedulerState::GetPeakMemoryUsage() in GetPeakMemoryUsage() function in tensorflow::grappler::SchedulerState