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

Fix internal attribute name used by grappler arithmetic optimizer (NFC)

Browse Source 
Attributes are supposed to match the regex "[a-z][a-z0-9_]+" according to
the documentation in op_def.proto

PiperOrigin-RevId: 259013399
This commit is contained in:
Mehdi Amini 2019-07-19 12:08:38 -07:00 committed by TensorFlower Gardener
parent 86ba7c7111
commit be42a1eb12
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc
View File

@ -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 <typename T>