Searched defs:reduce_sum (Results 1 – 10 of 10) sorted by relevance
33 CNodePtr CreateFusionNode(const FuncGraphPtr &graph, const CNodePtr &reduce_sum, const AnfNodePtr &… in CreateFusionNode()79 const AnfNodePtr &reduce_sum, const AnfNodePtr &input2) { in QuitFusion()122 VectorRef reduce_sum({prim::kPrimReduceSum, mul1}); in DefinePattern() local133 auto reduce_sum = node->cast<CNodePtr>(); in Process() local
20 reduce_sum = P.ReduceSum(keep_dims=True) variable
22 reduce_sum = P.ReduceSum() variable
21 reduce_sum = P.ReduceSum() variable
457 auto reduce_sum = func_graph->NewCNode(reduce_sum_inputs); in CreateDwReduceSum() local507 auto reduce_sum = func_graph->NewCNode(reduce_sum_inputs); in CreateDbReduceSum() local
31 ReduceSumInfoPtr reduce_sum; variable
91 reduce_sum = P.ReduceSum() variable
50 reduce_sum = P.ReduceSum() variable
34 reduce_sum = P.ReduceSum() variable
33 reduce_sum = P.ReduceSum() variable