Searched defs:v_node (Results 1 – 14 of 14) sorted by relevance
63 …std::vector<std::shared_ptr<Edge>> GetOriginalPrevEdges(const OperatorInfoPtr &v_node) { return in… in GetOriginalPrevEdges()80 …shared_ptr<Edge>> GetOriginalEdgeBetweenOperators(OperatorInfoPtr u_node, OperatorInfoPtr v_node) { in GetOriginalEdgeBetweenOperators()
62 void CostGraph::AddEdge(OperatorInfoPtr u_node, OperatorInfoPtr v_node, const EdgePtr &edge) { in AddEdge()
38 bool GetScalarValueFromNode(const ValueNodePtr &v_node, int64_t *v) { in GetScalarValueFromNode()
39 bool GetScalarValueFromNode(const ValueNodePtr &v_node, int64_t *v) { in GetScalarValueFromNode()
184 mindspore::ValueNodePtr v_node = std::make_shared<mindspore::ValueNode>(MakeValue(values)); in build_tuple_value_node() local612 mindspore::ValueNodePtr v_node = std::make_shared<mindspore::ValueNode>(prim_node); in BuildOneInputNoWeightNode() local657 mindspore::ValueNodePtr v_node = node_inputs[0]->cast<mindspore::ValueNodePtr>(); in BuildOneInputWithWeightNode() local
84 auto v_node = NewValueNode(value); in GenNewTensor() local
229 auto v_node = utils::cast<AnfNodePtr>((*equiv)[input_1_batchmm_sv_]); in CreateFlashAttentionNodePart1() local
108 auto v_node = PyNativeAlgo::Common::CreateValueNodeByValue(v); in GetNonTensorInput() local1994 auto v_node = PyNativeAlgo::Common::CreateValueNodeByValue(v); in GetInput() local
571 auto v_node = PyNativeAlgo::Common::CreateValueNodeByValue(grad_param->op_grad_info->out_value, in BuildKNode() local
37 void CreateTensorByConstantValue(const ValueNodePtr &v_node) { in CreateTensorByConstantValue()
676 auto v_node = input_node->cast<ValueNodePtr>(); in GetInputArgs() local
250 auto v_node = dyn_cast_ptr<ValueNode>(tuple_get_item->input(idx)); in SkipHookNodeInBackProp() local
640 auto v_node = cnode->forward().first; in ReplaceCNodeWithValueNode() local844 auto v_node = NewValueNode(v); in CreateValueNodeByValue() local
2047 auto v_node = NewValueNode(v); in GetPyExecuteFromValueSequence() local2066 auto v_node = NewValueNode(v); in GetPyExecuteFromValueSequence() local