OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-07 00:20:20 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix a tfprof bug. Throws an error when the flops cannot be calculated.

Browse Source 
PiperOrigin-RevId: 173702740
This commit is contained in:
A. Unique TensorFlower 2017-10-27 12:15:56 -07:00 committed by TensorFlower Gardener
parent 73155f56a3
commit 3d39b32b9a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/python/profiler/internal/flops_registry.py
View File

@ -373,6 +373,7 @@ def _max_pool_grad_flops(graph, node):
kernel_area = _list_product(kernel_shape)
orig_out_shape = graph_util.tensor_shape_from_node_def_name(graph,
node.input[1])
orig_out_shape.assert_is_fully_defined()
max_pool_ops = kernel_area * orig_out_shape.num_elements()
return ops.OpStats("flops", max_pool_ops + orig_out_shape.num_elements())