experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add memory usage details to the cost graph.

Browse Source 
PiperOrigin-RevId: 264921808
This commit is contained in:
Lluis-Miquel Munguia 2019-08-22 14:36:57 -07:00 committed by TensorFlower Gardener
parent 78299d8dbf
commit fbc490b2f0
1 changed files with 2 additions and 0 deletions

2
tensorflow/core/grappler/costs/analytical_cost_estimator.cc
View File

@ -61,6 +61,8 @@ void AddCostNode(ReadyNodeManager* node_manager, const OpContext& op_context,
node->set_compute_cost(node_costs.execution_time.asMicroSeconds().count());
node->set_compute_time(node_costs.compute_time.asMicroSeconds().count());
node->set_memory_time(node_costs.memory_time.asMicroSeconds().count());
node->set_temporary_memory_size(node_costs.temporary_memory);
node->set_persistent_memory_size(node_costs.persistent_memory);
node->set_inaccurate(node_costs.inaccurate);
for (const string& input : node_manager->GetCurrNode()->input()) {