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

Tweak comment about identity optimization

Browse Source
This commit is contained in:
Ben Barsdell 2019-05-01 11:42:35 -07:00
parent a9261f1243
commit bb196f949f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/grappler/optimizers/constant_folding.cc
View File

@ -2470,7 +2470,7 @@ bool ConstantFolding::IsReductionSimplifiableToIdentity(
bool ConstantFolding::ReplaceReductionWithIdentity(NodeDef* node) const { bool ConstantFolding::ReplaceReductionWithIdentity(NodeDef* node) const {
// Replace the reduction node with an identity node, that can be further // Replace the reduction node with an identity node, that can be further
// optimized by the model pruner. // optimized by other passes.
DataType output_type; DataType output_type;
if (node->attr().count("T") != 0) { if (node->attr().count("T") != 0) {
output_type = node->attr().at("T").type(); output_type = node->attr().at("T").type();