Home
last modified time | relevance | path

Searched refs:peak_usage (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/grappler/
Dcost_analyzer.py73 peak_usage = cluster.DeterminePeakMemoryUsage(item)
75 for device, snapshot in peak_usage.items():
76 peak_usage = snapshot[0]
78 peak_usage) + " bytes\n"
Dcluster_test.py83 peak_usage = snapshot[0]
84 self.assertEqual(52, peak_usage)
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc66 for (const auto& peak_usage : peak_usage_) { in GetWorstCaseMemoryUsage() local
67 worst_case = std::max(worst_case, peak_usage.second.used_memory); in GetWorstCaseMemoryUsage()