diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-20 16:51:05 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-20 16:51:10 -0700 |
| commit | 479d43f8071004240e28e3c65471a7a3f37bc605 (patch) | |
| tree | abf83d0cf2fb07b9dbe269c18754ff9902b0e9f2 | |
| parent | 8741006018326350467fe86785d98963ff9e983e (diff) | |
| parent | aede46743b0256ef382f23b0db0370fe0777116d (diff) | |
Merge pull request #20993 from joeyearsley:patch-5
PiperOrigin-RevId: 205473909
| -rw-r--r-- | tensorflow/tools/graph_transforms/fold_old_batch_norms.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/graph_transforms/fold_old_batch_norms.cc b/tensorflow/tools/graph_transforms/fold_old_batch_norms.cc index f1d361e07d..156636ab82 100644 --- a/tensorflow/tools/graph_transforms/fold_old_batch_norms.cc +++ b/tensorflow/tools/graph_transforms/fold_old_batch_norms.cc @@ -159,7 +159,7 @@ Status FuseScaleOffsetToConvWeights(const std::vector<float>& scale_values, NodeDef bias_add_node; bias_add_node.set_op("BiasAdd"); bias_add_node.set_name(conv_output_name); - if (!conv_node.attr().count("data_format")) { + if (conv_node.attr().count("data_format") > 0) { CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); } CopyNodeAttr(conv_node, "T", "T", &bias_add_node); |