From be42a1eb1202792cbbb18941384cfacc7ef5dd67 Mon Sep 17 00:00:00 2001 From: Mehdi Amini Date: Fri, 19 Jul 2019 12:08:38 -0700 Subject: [PATCH] Fix internal attribute name used by grappler arithmetic optimizer (NFC) Attributes are supposed to match the regex "[a-z][a-z0-9_]+" according to the documentation in op_def.proto PiperOrigin-RevId: 259013399 --- tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc index 273460050fc..ebf704c0718 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc @@ -62,9 +62,9 @@ namespace { // Mark nodes created or optimized by a stage with a tag. constexpr char kAddOpsRewriteTag[] = - "_grappler:ArithmeticOptimizer:AddOpsRewriteStage"; + "_grappler_ArithmeticOptimizer_AddOpsRewriteStage"; constexpr char kMinimizeBroadcastsTag[] = - "_grappler:ArithmeticOptimizer:MinimizeBroadcasts"; + "_grappler_ArithmeticOptimizer_MinimizeBroadcasts"; // Extract values from a Const op to `values`. Returns true if succeeds. template