diff options
author | Joe Yearsley <josephelliotyearsley@gmail.com> | 2018-07-20 15:00:29 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-07-20 15:00:29 +0100 |
commit | aede46743b0256ef382f23b0db0370fe0777116d (patch) | |
tree | abcf39e8ff5e198b3a06a80ae88dee04c0e354c9 /tensorflow/tools/graph_transforms | |
parent | a641057c719aa95175a9ad3f9e26044f0c31416e (diff) |
Update fold_old_batch_norms.cc
Fixed my previous fix.
Diffstat (limited to 'tensorflow/tools/graph_transforms')
-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); |