From 2087cc7e33038d54181c97470d9d21a86024857c Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 1 Aug 2018 13:54:06 -0700 Subject: 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 --- tensorflow/core/grappler/costs/virtual_scheduler.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 GetPeakMemoryUsage() const; - protected: const std::unordered_map* 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( -- cgit v1.2.3