Home
last modified time | relevance | path

Searched refs:ValuePtrList (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dcc_implementations.h31 using ValuePtrList = std::vector<ValuePtr>; variable
35 ValuePtr ScalarAdd(const ValuePtrList &list);
36 ValuePtr ScalarSub(const ValuePtrList &list);
37 ValuePtr ScalarMul(const ValuePtrList &list);
38 ValuePtr ScalarDiv(const ValuePtrList &list);
39 ValuePtr ScalarMod(const ValuePtrList &list);
40 ValuePtr ScalarPow(const ValuePtrList &list);
41 ValuePtr ScalarFloordiv(const ValuePtrList &list);
42 ValuePtr ScalarUAdd(const ValuePtrList &list);
43 ValuePtr ScalarUSub(const ValuePtrList &list);
[all …]
Dcc_implementations.cc199 …ValuePtr Scalar##op_t(const ValuePtrList &list) { …
271 …ValuePtr Scalar##op_t(const ValuePtrList &list) { … in SCALAR_OP()
347 ValuePtr ScalarUAdd(const ValuePtrList &list) {
356 ValuePtr ScalarUSub(const ValuePtrList &list) { in ScalarUSub()
379 ValuePtr ScalarLog(const ValuePtrList &list) { in ScalarLog()
398 ValuePtr BoolNot(const ValuePtrList &list) { in BoolNot()
414 ValuePtr BoolAnd(const ValuePtrList &list) { in BoolAnd()
433 ValuePtr BoolOr(const ValuePtrList &list) { in BoolOr()
452 ValuePtr BoolEq(const ValuePtrList &list) { in BoolEq()
/third_party/mindspore/tests/ut/cpp/operator/
Dcc_implementations_test.cc33 ValuePtrList list; in TEST_F()
71 ValuePtrList list; in TEST_F()
109 ValuePtrList list; in TEST_F()
172 ValuePtrList list; in TEST_F()
215 ValuePtrList list; in TEST_F()
253 ValuePtrList list; in TEST_F()
260 ValuePtrList list; in TEST_F()
267 ValuePtrList list; in TEST_F()
274 ValuePtrList list; in TEST_F()
297 ValuePtrList list; in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ad/
Dkpynative.h37 …virtual bool KPynativeWithFProp(const CNodePtr &c_node, const ValuePtrList &op_args, const ValuePt…
50 …zeBPropFuncGraph(const FuncGraphPtr &bprop_fg, const CNodePtr &c_node, const ValuePtrList &op_args,
80 bool GradPynativeOp(const KPynativeCellPtr &k_cell, const CNodePtr &c_node, const ValuePtrList &op_…
89 …ativeWithBProp(const KPynativeCellPtr &k_cell, const CNodePtr &c_node, const ValuePtrList &op_args,
Dprim_bprop_optimizer.h107 void TryFreeArgsValue(const ValuePtrList &op_args, const ValuePtr &out);
117 …void TryFreeOneValue(const ValuePtrList &op_args, const std::vector<ParamUsingInfo> &param_info_ve…
143 …zeBPropFuncGraph(const FuncGraphPtr &bprop_fg, const CNodePtr &c_node, const ValuePtrList &op_args,
157 …void ArgsToAbs(const PrimitivePtr &prim, const ValuePtrList &op_args, abstract::AbstractBasePtrLis…
171 …FuncGraphPtr GetOptBpropFromCache(const FuncGraphPtr &bprop_fg, const ValuePtrList &op_args, const…
174 …FuncGraphPtr GenSpecOptBprop(const FuncGraphPtr &bprop_fg, const ValuePtrList &op_args, const Valu…
Dkpynative.cc196 …PynativeAdjoint(const FuncGraphPtr &tape, const ValuePtrList &op_args, const ValuePtr &out, const … in PynativeAdjoint()
202 const ValuePtrList &op_args() const { return op_args_; } in op_args()
240 const ValuePtrList op_args_;
263 …std::make_shared<PynativeAdjoint>(tape_, ValuePtrList{}, input_param_values[i], FuncGraphPtr(nullp… in KPynativeCellImpl()
268 bool KPynativeOp(const CNodePtr &cnode, const ValuePtrList &op_args, const ValuePtr &out);
269 bool KPynativeWithBProp(const CNodePtr &cnode, const ValuePtrList &op_args, const ValuePtr &out,
271 bool KPynativeWithFProp(const CNodePtr &c_node, const ValuePtrList &op_args, const ValuePtr &out,
294 bool BuildAdjoint(const CNodePtr &cnode, const ValuePtrList &op_args, const ValuePtr &out,
297 void BuildAdjointForInput(const CNodePtr &cnode, const ValuePtrList &op_args);
378 bool GradPynativeOp(const KPynativeCellPtr &k_cell, const CNodePtr &cnode, const ValuePtrList &op_a… in GradPynativeOp()
[all …]
Dprim_bprop_optimizer.cc24 void PrimBpropOptGraphLevel2Info::TryFreeArgsValue(const ValuePtrList &op_args, const ValuePtr &out… in TryFreeArgsValue()
32 ValuePtrList new_args(op_args); in TryFreeArgsValue()
37 void PrimBpropOptGraphLevel2Info::TryFreeOneValue(const ValuePtrList &op_args, in TryFreeOneValue()
179 … const ValuePtrList &op_args, const ValuePtr &out) { in OptimizeBPropFuncGraph()
206 …PrimBpropOptimizer::GetOptBpropFromCache(const FuncGraphPtr &bprop_fg, const ValuePtrList &op_args, in GetOptBpropFromCache()
243 FuncGraphPtr PrimBpropOptimizer::GenSpecOptBprop(const FuncGraphPtr &bprop_fg, const ValuePtrList &… in GenSpecOptBprop()
342 void PrimBpropOptimizer::ArgsToAbs(const PrimitivePtr &prim, const ValuePtrList &op_args, in ArgsToAbs()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconst_to_attr_strided_slice_grad.cc32 bool GetStridesValues(const CNodePtr &strided_slice_grad, ValuePtrList *strides_values) { in GetStridesValues()
61 bool CheckValues(const ValuePtrList &strides_values) { in CheckValues()
132 ValuePtrList strides_values; in Process()
/third_party/mindspore/mindspore/core/ops/
Dcustom.cc36 ValuePtrList value_ptr_list; in set_attr()
46 auto value_ptr_list = GetValue<ValuePtrList>(this->GetAttr(kAttr)); in get_attr()
Dneighborexchange.cc48 ValuePtrList attr_shapes; in CheckAttr()
54 attr_shapes = GetValue<ValuePtrList>(attr); in CheckAttr()
116 auto send_shapes = GetValue<ValuePtrList>(primitive->GetAttr(kSendShapes)); in Check()
Dreduce_sum.cc47 ValuePtrList axis_items = axis_ptr_list; in InferImplReduceFuncCalShape()
48 ValuePtrList::iterator it; in InferImplReduceFuncCalShape()
Ddropout_gen_mask.cc33 ShapeVector CalDynamicOutputShape(const ValuePtrList value_list) { in CalDynamicOutputShape()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/
Dprim.h117 using ValuePtrList = std::vector<ValuePtr>; variable
118 using PrimitiveImpl = ValuePtr (*)(const ValuePtrList &);
144 ValuePtr RunImpl(const ValuePtrList &args) const;
Dprim.cc712 ValuePtrList value_list; in EvalPrim()
757 ValuePtr UniformPrimEvaluator::RunImpl(const ValuePtrList &args) const { in RunImpl()
/third_party/mindspore/mindspore/core/ir/
Dvalue.h39 explicit ValueSequeue(const ValuePtrList &elements) : elements_(elements) { in ValueSequeue()
61 const ValuePtrList &value() const { return elements_; } in value()
68 ValuePtrList elements_;
Danf.h60 using ValuePtrList = std::vector<ValuePtr>; variable
/third_party/mindspore/mindspore/core/abstract/
Dprim_maths.cc90 ValuePtrList axis_items = axis_ptr_list; in InferImplReduceFuncCalShape()
91 ValuePtrList::iterator it; in InferImplReduceFuncCalShape()
92 ValuePtrList::reverse_iterator it_re; in InferImplReduceFuncCalShape()
/third_party/mindspore/tests/ut/cpp/optimizer/ad/
Dkpynative_test.cc94 ValuePtrList args; in BuildBpropFuncGraph()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/parse/
Dresolve.cc231 const auto &vec = GetValue<ValuePtrList>(elem); in IsAllFuncInValueSequence()
272 const auto &value_vec = GetValue<ValuePtrList>(value_node->value()); in TransformVectorFuncValueNode()
/third_party/mindspore/mindspore/ccsrc/pipeline/pynative/
Dpynative_execute.h206 …void MakeCNodeForMsFunction(const FuncGraphPtr &ms_func_graph, const py::args &args, ValuePtrList …
227 std::vector<AnfNodePtr> *inputs, ValuePtrList *weights_args);
Dpynative_execute.cc1590 ValuePtrList input_args; in DoOpGrad()
1627 ValuePtrList *input_values, CNodePtr *ms_function_cnode) { in MakeCNodeForMsFunction()
1690 ValuePtrList input_values; in MakeAdjointForMsFunction()
2350 ValuePtrList input_args; in CreateMakeTupleNodeForMultiOut()
2882 std::vector<AnfNodePtr> *inputs, ValuePtrList *weights_args) { in DoParameterReplace()
2960 ValuePtrList weights_args; in MakeNestedCnode()
2979 ValuePtrList input_args; in MakeNestedCnode()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dincorporate_getitem.h249 ValuePtrList shrunk_inputs; in ShrinkUnsedOutput()