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.h102 struct DeviceState { struct
149 DeviceState() { in DeviceState() argument
364 const std::unordered_map<string, DeviceState>* GetDeviceStates() const { in GetDeviceStates()
396 std::unordered_map<string, DeviceState>* GetMutableDeviceState() { in GetMutableDeviceState()
420 std::unordered_map<string, DeviceState> device_;
506 const std::unordered_map<string, DeviceState>* GetDeviceStates() const { in GetDeviceStates()
Dvirtual_scheduler.cc95 DeviceState* device) { in UpdateDeviceAnnotationState()
1282 const DeviceState& state = device.second; in GetPeakMemoryUsage()
1293 const DeviceState& state = device.second; in GetPersistentMemoryUsage()
Dvirtual_scheduler_test.cc2352 DeviceState::NodePairHash>& mem_usage_snapshot) { in ValidateMemoryUsageSnapshot()