Skip to content

Commit

Permalink
post ops of reduce done as subgraph as WA
Browse files Browse the repository at this point in the history
  • Loading branch information
chenhu-wang committed May 27, 2024
1 parent 883211e commit 83eaa01
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -402,7 +402,7 @@ bool isSuitableChildForFusingSumActivation(const std::shared_ptr<const Node> &no
return SupportsFusingWithConvolution_SumActivation(node);
}
bool isSuitableReduceChild(const std::shared_ptr<const Node> &node, const int channelAxis = DEFAULT_AXIS) {
return isSuitableChildForFusingSimple(node, channelAxis);
return node->get_output_element_type(0) == ov::element::f32 && isSuitableChildForFusingSimple(node, channelAxis);
}
bool isSuitableMatMulWithConstantPath(const std::shared_ptr<Node>& node) {
return ov::is_type<ov::opset1::MatMul>(node) &&
Expand Down

0 comments on commit 83eaa01

Please sign in to comment.