Searched refs:kUMonad (Results 1 – 15 of 15) sorted by relevance
97 auto u = NewValueNode(kUMonad); in KeepExecOrder()98 u->set_abstract(kUMonad->ToAbstract()); in KeepExecOrder()
234 new_node = NewValueNode(kUMonad)->cast<AnfNodePtr>(); in ProcessIndex()235 new_node->set_abstract(kUMonad->ToAbstract()); in ProcessIndex()
385 auto u = NewValueNode(kUMonad); in InsertUpdateState()386 u->set_abstract(kUMonad->ToAbstract()); in InsertUpdateState()389 update_state_cnode->set_abstract(kUMonad->ToAbstract()); in InsertUpdateState()
646 AddMonadParameters(branches, "u", kUMonad->ToAbstract()); in SetupEffectBranches()1016 AddMonadArgument(caller, kUMonad); in AddMonadForCaller()1038 if ((monad == kUMonad) && cnode->size() > 1 && HasAbstractIOMonad(cnode->inputs().back())) { in AddMonadArgument()1211 auto u_node = NewValueNode(kUMonad); in HandleOuterNode()1212 u_node->set_abstract(kUMonad->ToAbstract()); in HandleOuterNode()1421 u_ = NewValueNode(kUMonad); in GetUniverse()1422 u_->set_abstract(kUMonad->ToAbstract()); in GetUniverse()1424 u_ = AddMonadParameter(func_graph_, "u", kUMonad->ToAbstract()); in GetUniverse()
241 auto monad = NewValueNode(kUMonad); in GetFirstMonad()242 monad->set_abstract(kUMonad->ToAbstract()); in GetFirstMonad()
377 auto u = NewValueNode(kUMonad); in NewRecomputedNode()378 u->set_abstract(kUMonad->ToAbstract()); in NewRecomputedNode()
379 auto u = NewValueNode(kUMonad); in AdjustForAutoMonad()380 u->set_abstract(kUMonad->ToAbstract()); in AdjustForAutoMonad()485 extra_node = NewValueNode(kUMonad); in BuildOutput()
250 auto const_u_monad = NewValueNode(kUMonad); in AdjustAllReduceInputWithLoad()251 const_u_monad->set_abstract(kUMonad->ToAbstract()); in AdjustAllReduceInputWithLoad()
307 const ValuePtr kUMonad = std::make_shared<UMonad>(); variable
284 extern const ValuePtr kUMonad;
660 auto monad_abs = kUMonad->ToAbstract(); in ObtainValueNodeInMonadForm()661 auto new_value_node = NewValueNode(kUMonad); in ObtainValueNodeInMonadForm()898 cnode_ptr->set_abstract(kUMonad->ToAbstract()); in SetCNodeAbastract()
1335 monad_value_ = kernel_graph_->NewValueNode(kUMonad->ToAbstract(), kUMonad); in GetMonadValue()1559 auto monad = graph_->NewValueNode(kUMonad->ToAbstract(), kUMonad); in InsertLabelSet()
173 return kg->NewValueNode(kUMonad->ToAbstract(), kUMonad); in MakeMonadValueNode()
374 return kUMonad->ToAbstract(); in MakeMonadAbstract()
782 explicit AbstractUMonad(const ValuePtr &value = kUMonad) : AbstractMonad(value, kUMonadType) {} in AbstractMonad()