diff options
author | 2018-10-08 23:12:08 -0700 | |
---|---|---|
committer | 2018-10-08 23:16:17 -0700 | |
commit | e27ee15fa45a5f4e43e10ed1fe0eb3a1feb4253a (patch) | |
tree | 2588e0531141c95d8c443fa4923d2df20b4970fc /tensorflow/core/grappler/optimizers/BUILD | |
parent | d1f0494b89a31298df7743018c0a3fa388ac16a2 (diff) |
Refactor CalculateOutputSize() from VirtualScheduler protected member function to utils; Refactor EstimateSize() from memory_optimizer.cc to utils; some small changes for readability improvement
PiperOrigin-RevId: 216307257
Diffstat (limited to 'tensorflow/core/grappler/optimizers/BUILD')
-rw-r--r-- | tensorflow/core/grappler/optimizers/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/BUILD b/tensorflow/core/grappler/optimizers/BUILD index c708f84948..e898377ded 100644 --- a/tensorflow/core/grappler/optimizers/BUILD +++ b/tensorflow/core/grappler/optimizers/BUILD @@ -423,6 +423,7 @@ cc_library( "//tensorflow/core/grappler/clusters:virtual_cluster", "//tensorflow/core/grappler/costs:graph_memory", "//tensorflow/core/grappler/costs:graph_properties", + "//tensorflow/core/grappler/costs:utils", "//tensorflow/core/grappler/utils:topological_sort", "//tensorflow/core/grappler/utils:traversal", ], |