diff options
author | 2018-05-23 16:45:26 -0700 | |
---|---|---|
committer | 2018-05-23 16:48:01 -0700 | |
commit | dac1f124020234fe24e8893a981b15395d0c6de8 (patch) | |
tree | 54b3b50356fba1d4446f43a4fc50cd1202e21828 /tensorflow/tools/graph_transforms | |
parent | 4ba9e8eed9dfe0727db000bdd8be5384f39e6bd9 (diff) |
Simplify the remapper code and added support for non scalar mean, variance, scale and offset.
PiperOrigin-RevId: 197812268
Diffstat (limited to 'tensorflow/tools/graph_transforms')
-rw-r--r-- | tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc b/tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc index 7651a03fe5..435f46c107 100644 --- a/tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc +++ b/tensorflow/tools/graph_transforms/fold_old_batch_norms_test.cc @@ -191,7 +191,7 @@ class FoldOldBatchNormsTest : public ::testing::Test { std::vector<Tensor> fused_outputs; TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); - test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); + test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 2e-5); for (const NodeDef& node : fused_graph_def.node()) { EXPECT_NE("FusedBatchNorm", node.op()); |