|
|
@ -58,9 +58,7 @@ void Conv2DBackpropEltwiseFusionPass::MatchSingleFusionPattern(const session::Ke
|
|
|
|
}
|
|
|
|
}
|
|
|
|
auto cnode = node->cast<CNodePtr>();
|
|
|
|
auto cnode = node->cast<CNodePtr>();
|
|
|
|
MS_EXCEPTION_IF_NULL(cnode);
|
|
|
|
MS_EXCEPTION_IF_NULL(cnode);
|
|
|
|
if (AnfAlgo::GetKernelType(cnode) == KernelType::TBE_KERNEL &&
|
|
|
|
if (AnfAlgo::GetCNodeName(cnode) == kReluGradV2OpName) {
|
|
|
|
AnfAlgo::GetFusionType(cnode) == kernel::FusionType::ELEMWISE &&
|
|
|
|
|
|
|
|
(cnode->inputs().size() == ELTWISE_INPUT_SIZE || cnode->inputs().size() == ELTWISE_DOUBLE_IN_INPUT_SIZE)) {
|
|
|
|
|
|
|
|
MatchConv2DBackpropInputEltwise(cnode, kernel_graph, candidate_fusion);
|
|
|
|
MatchConv2DBackpropInputEltwise(cnode, kernel_graph, candidate_fusion);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|