Searched defs:k_node (Results 1 – 5 of 5) sorted by relevance
48 void set_k_node(const AnfNodePtr &k_node) { k_node_ = k_node; } in set_k_node()49 AnfNodePtr k_node() const { return k_node_.lock(); } in k_node() function
110 auto k_node = auto_grad_meta_data->k_node(); in IsConstant() local251 AnfNodePtr k_node = nullptr; in KPynativeOp() local291 AnfNodePtr k_node = nullptr; in KPynativeWithFProp() local421 … const VariablePtr &variable_adjoint, const AnfNodePtr &k_node, in ConstructBpropGraphInput()484 auto k_node = auto_grad_meta_data->k_node(); in BuildKNodeForCNodeInput() local504 auto k_node = ad_param_->tape_->FuncGraph::NewCNode(inputs); in BuildKNodeForCNodeInput() local532 void IrGrad::SetKNodeInfo(const ValuePtr &value, const AnfNodePtr &k_node, const AbstractBasePtr &o… in SetKNodeInfo()567 auto k_node = ad_param()->tape_->FuncGraph::NewCNode(node_list); in BuildKNode() local
718 auto k_node = GetKnode(prim, cnode, *cnode_inputs, jit_by_value); in GradCNode() local796 auto k_node = ad_param_->tape_->FuncGraph::NewCNode(inputs); in BuildKNodeForMakeTuple() local866 auto k_node = ad_param_->tape_->FuncGraph::NewCNode(inputs); in BuildKNodeForTupleGetItem() local
314 AnfNodePtr k_node() const { return k_node_; } in k_node() function315 void set_k_node(const AnfNodePtr &k_node) { k_node_ = k_node; } in set_k_node()
227 auto k_node = utils::cast<AnfNodePtr>((*equiv)[input_1_batchmm_qk_]); in CreateFlashAttentionNodePart1() local