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

Check for Any/All reduction explicitly

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

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

@ -2474,9 +2474,10 @@ bool ConstantFolding::ReplaceReductionWithIdentity(NodeDef* node) const {
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();
} else { } else if (IsAny(*node) || IsAll(*node)) {
// This is an 'any' or 'all' reduction. The output is always boolean.
output_type = DT_BOOL; output_type = DT_BOOL;
} else {
return false;
} }
node->set_op("Identity"); node->set_op("Identity");
node->clear_attr(); node->clear_attr();