Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.h87 struct DeviceState { struct
119 DeviceState() { in DeviceState() argument
294 const std::unordered_map<string, DeviceState>* GetDeviceStates() const { in GetDeviceStates()
332 std::unordered_map<string, DeviceState> device_;
Dvirtual_scheduler.cc1083 const DeviceState& state = device.second; in GetPeakMemoryUsage()
Dvirtual_scheduler_test.cc1678 DeviceState::NodePairHash>& mem_usage_snapshot) { in ValidateMemoryUsageSnapshot()