Searched refs:acc_node (Results 1 – 2 of 2) sorted by relevance
108 const CNodePtr &acc_node = inplace_node[acc_index].node->cast<CNodePtr>(); in ExistDependencyFromAcc2Cover() local110 MS_EXCEPTION_IF_NULL(acc_node); in ExistDependencyFromAcc2Cover()111 return ExistRoute(acc_node, cover_node); in ExistDependencyFromAcc2Cover()178 const CNodePtr &acc_node = inplace_node->at(acc_index).node->cast<CNodePtr>(); in CheckInplaceNodeInputs() local180 MS_EXCEPTION_IF_NULL(acc_node); in CheckInplaceNodeInputs()181 const auto &acc_input = acc_node->input(1)->cast<CNodePtr>(); in CheckInplaceNodeInputs()191 auto new_inplace_node = graph->NewCNode({acc_node->input(0), new_input, acc_node->input(2)}); in CheckInplaceNodeInputs()193 new_inplace_node->set_abstract(acc_node->abstract()); in CheckInplaceNodeInputs()194 CopyKernelInfo(acc_node, new_inplace_node); in CheckInplaceNodeInputs()197 manager->Replace(acc_node, new_inplace_node); in CheckInplaceNodeInputs()[all …]
65 gpir_node *acc_node = gpir_instr_get_the_other_acc_node(instr, slot); in gpir_instr_check_acc_same_op() local68 if (acc_node && acc_node != node && in gpir_instr_check_acc_same_op()69 !gpir_codegen_acc_same_op(node->op, acc_node->op)) in gpir_instr_check_acc_same_op()78 gpir_node *acc_node = gpir_instr_get_the_other_acc_node(instr, node->sched.pos); in gpir_instr_get_consume_slot() local79 if (acc_node) in gpir_instr_get_consume_slot()