Tweak comment about identity optimization

This commit is contained in:
Ben Barsdell 2019-05-01 11:42:35 -07:00
parent a9261f1243
commit bb196f949f

View File

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