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

Copy exponential_avg_factor attr in grappler

Browse Source
This commit is contained in:
ShengYang1 2020-04-08 14:07:24 +08:00
parent d4d23502bf
commit 08cd923c40
1 changed files with 1 additions and 0 deletions

1
tensorflow/core/grappler/optimizers/remapper.cc
View File

@ -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") {