diff options
| author |  Yao Zhang <yaozhang@google.com> | 2017-06-13 19:47:59 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-13 22:10:58 -0700 |
| commit | 6ffa51f1e0e76c87ee2164a8d421279768372501 (patch) | |
| tree | ad2c4ee07d5fbed07113c3bbda41a0c4dfc683a0 /tensorflow/core/grappler/costs/op_level_cost_estimator.cc | |
| parent | 838571b0a05f38db0715a149b22094cc1cd6ff75 (diff) | |
Add a standalone cost analysis tool. Improved logging.
PiperOrigin-RevId: 158933442
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.cc')
| -rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc index d8b8a12eb2..ba6686e7df 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.cc +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.cc @@ -314,6 +314,8 @@ std::pair<double, double> OpLevelCostEstimator::GetDeviceInfo( bandwidth = 100; } } + VLOG(1) << "Device: " << device.type() << " GFLOPS: " << gflops + << " Bandwidth: " << bandwidth; return std::make_pair(gflops, bandwidth); } @@ -461,7 +463,7 @@ int64 OpLevelCostEstimator::CountConv2DOperations( ops *= conv_dims.kx * conv_dims.ky; ops *= conv_dims.iz * conv_dims.oz; ops *= kOpsPerMac; - VLOG(1) << "Operations for Conv2D" << ops; + VLOG(1) << "Operations for Conv2D " << ops; if (conv_info != nullptr) { *conv_info = conv_dims; @@ -679,7 +681,7 @@ int64 OpLevelCostEstimator::CountConv2DBackPropInputOperations( ops *= conv_dims.iz * conv_dims.oz; ops *= kOpsPerMac; - VLOG(1) << "Operations for Conv2DBackPropInput" << ops; + VLOG(1) << "Operations for Conv2DBackPropInput " << ops; if (returned_conv_dims != nullptr) { *returned_conv_dims = conv_dims; |