From 08cd923c406a20aa019344e081a74f03ad2ebd0f Mon Sep 17 00:00:00 2001 From: ShengYang1 Date: Wed, 8 Apr 2020 14:07:24 +0800 Subject: [PATCH] Copy exponential_avg_factor attr in grappler --- tensorflow/core/grappler/optimizers/remapper.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/core/grappler/optimizers/remapper.cc b/tensorflow/core/grappler/optimizers/remapper.cc index fd8c7a0af12..6f94f66cb00 100644 --- a/tensorflow/core/grappler/optimizers/remapper.cc +++ b/tensorflow/core/grappler/optimizers/remapper.cc @@ -885,6 +885,7 @@ void CopyFusedBatchNormAttributes(const NodeDef& fused_batch_norm, (*attr)["is_training"] = src_attr.at("is_training"); (*attr)["data_format"] = src_attr.at("data_format"); (*attr)["epsilon"] = src_attr.at("epsilon"); + (*attr)["exponential_avg_factor"] = src_attr.at("exponential_avg_factor"); // FusedBatchNormV2 and V3 have an extra type parameter. if (fused_batch_norm.op() != "FusedBatchNorm") {