diff options
author | 2018-08-01 13:54:06 -0700 | |
---|---|---|
committer | 2018-08-01 13:57:55 -0700 | |
commit | 2087cc7e33038d54181c97470d9d21a86024857c (patch) | |
tree | c87126cfba31df40e44be9a85cfd43d415e9a72d | |
parent | 4963bcf071dea91bb1fca38d1c281d9d76a7dfa3 (diff) |
Move GetDeviceStates() and GetNodeState() functions from protected to public, so that the user can have access to more detailed results from VirtualScheduler.
PiperOrigin-RevId: 206986812
-rw-r--r-- | tensorflow/core/grappler/costs/virtual_scheduler.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/costs/virtual_scheduler.h b/tensorflow/core/grappler/costs/virtual_scheduler.h index 34d48819ac..353ca6f071 100644 --- a/tensorflow/core/grappler/costs/virtual_scheduler.h +++ b/tensorflow/core/grappler/costs/virtual_scheduler.h @@ -275,7 +275,6 @@ class VirtualScheduler { // Return per device peak memory usage. const std::unordered_map<string, int64> GetPeakMemoryUsage() const; - protected: const std::unordered_map<string, DeviceState>* GetDeviceStates() const { return &device_; } @@ -283,6 +282,7 @@ class VirtualScheduler { return &node_map_; } + protected: // Returns the size of output at port_num (unit: bytes). A special case is // port_num -1, which is for control dependency and assumed to be 4 bytes. int64 CalculateOutputSize( |