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

Merge pull request #31993 from ppwwyyxx:patch-19

Browse Source 
PiperOrigin-RevId: 269556368
This commit is contained in:
TensorFlower Gardener 2019-09-17 06:45:12 -07:00
commit 21a7e525a1
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/ops/math_ops.py
View File

@ -2871,6 +2871,7 @@ def _calc_mat_mul_flops(graph, node):
@ops.RegisterStatistics("BatchMatMul", "flops") @ops.RegisterStatistics("BatchMatMul", "flops")
@ops.RegisterStatistics("BatchMatMulV2", "flops")
def _calc_batch_mat_mul_flops(graph, node): def _calc_batch_mat_mul_flops(graph, node):
"""Calculates the compute resources needed for BatchMatMul.""" """Calculates the compute resources needed for BatchMatMul."""
transpose_a = node.attr["transpose_a"].b transpose_a = node.attr["transpose_a"].b