| /third_party/mindspore/tests/ut/cpp/operator/ |
| D | composite_test.cc | 36 using AbstractTuple = abstract::AbstractTuple; typedef
98 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
123 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
155 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
162 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
181 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
188 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
207 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
214 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
233 auto tuple_tensor = std::make_shared<AbstractTuple>(eles); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.cc | 160 auto x_reduce_idx = std::make_shared<AbstractTuple>(abs_list_x); in BroadcastGradientArgsDiff()
161 auto y_reduce_idx = std::make_shared<AbstractTuple>(abs_list_y); in BroadcastGradientArgsDiff()
166 return std::make_shared<AbstractTuple>(elem_list); in BroadcastGradientArgsDiff()
223 return std::make_shared<AbstractTuple>(values); in DoInferReduceShape()
247 return std::make_shared<AbstractTuple>(values); in DoInferReduceShape()
258 auto arg_x = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplBroadcastGradientArgs()
259 auto arg_y = CheckArg<AbstractTuple>(op_name, args_spec_list, 1); in InferImplBroadcastGradientArgs()
273 auto x_reduce_idx = std::make_shared<AbstractTuple>(empty_list); in InferImplBroadcastGradientArgs()
274 auto y_reduce_idx = std::make_shared<AbstractTuple>(empty_list); in InferImplBroadcastGradientArgs()
280 return std::make_shared<AbstractTuple>(elem_list); in InferImplBroadcastGradientArgs()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | graph_transform.cc | 43 if (abs->isa<abstract::AbstractTuple>()) { in CNodeHasTupleInput()
53 if (param->abstract()->isa<abstract::AbstractTuple>()) { in FuncGraphHasTupleInput()
67 if (elements[i]->isa<abstract::AbstractTuple>()) { in TransformTupleArgument()
87 if (abs->isa<abstract::AbstractTuple>()) { in TransformCallGraph()
110 if (abs->isa<abstract::AbstractTuple>()) { in TransformPartial()
132 if (abs->isa<abstract::AbstractTuple>()) { in TransformSwitchCall()
|
| D | clean.cc | 38 using mindspore::abstract::AbstractTuple;
58 return std::make_shared<AbstractTuple>(baselist); in Reabs()
66 return std::make_shared<AbstractTuple>(baselist); in Reabs()
79 return std::make_shared<AbstractTuple>(abs_list->elements()); in AdaptAbs()
85 return std::make_shared<AbstractTuple>(abstract_list); in AdaptAbs()
92 return std::make_shared<AbstractTuple>(abstract_list); in AdaptAbs()
545 if (!param_abs->isa<AbstractTuple>()) { in ExpandTuplesP()
552 auto abs_tuple = dyn_cast<AbstractTuple>(param_abs); in ExpandTuplesP()
581 if (abstract_tag->element()->isa<AbstractTuple>()) { in ExpandTuplesC()
586 if (!input_abs->isa<AbstractTuple>()) { in ExpandTuplesC()
[all …]
|
| D | graph_transform.h | 57 if (item->isa<abstract::AbstractTuple>()) { in GenerateTupleParams()
79 if (abs != nullptr && abs->isa<abstract::AbstractTuple>()) { in TransformGraphParam()
|
| /third_party/mindspore/tests/ut/cpp/abstract/ |
| D | utils_test.cc | 42 …AbstractBasePtr t1 = std::make_shared<AbstractTuple>(AbstractBasePtrList({abs_s1, abs_s_anything})… in TEST_F()
43 …AbstractBasePtr t2 = std::make_shared<AbstractTuple>(AbstractBasePtrList({abs_s1, abs_s_anything})… in TEST_F()
44 …AbstractBasePtr t3 = std::make_shared<AbstractTuple>(AbstractBasePtrList({abs_s_anything, abs_s_an… in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | rewrite_output_shape.cc | 40 mt_node->set_abstract(std::make_shared<abstract::AbstractTuple>(abstracts)); in Run()
53 if (node->abstract()->isa<abstract::AbstractTuple>()) { in Process()
60 node->set_abstract(std::make_shared<abstract::AbstractTuple>(node_abstracts)); in Process()
88 auto abs_tuple = dyn_cast<abstract::AbstractTuple>(output->abstract()); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | add_input_to_output.cc | 47 if (origin_abstract->isa<abstract::AbstractTuple>()) { in AddOutputs()
48 auto origin_abstract_tuple = dyn_cast<abstract::AbstractTuple>(origin_abstract); in AddOutputs()
73 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in AddOutputs()
107 auto new_abstract_tuple = dyn_cast<abstract::AbstractTuple>(cnode->abstract()); in Process()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | addn.cc | 30 …if (!input_args[0]->isa<abstract::AbstractTuple>() && !input_args[0]->isa<abstract::AbstractList>(… in AddNInferShape()
33 auto elements = input_args[0]->isa<abstract::AbstractTuple>() in AddNInferShape()
68 …if (!input_args[0]->isa<abstract::AbstractTuple>() && !input_args[0]->isa<abstract::AbstractList>(… in AddNInferType()
71 auto elements = input_args[0]->isa<abstract::AbstractTuple>() in AddNInferType()
|
| D | dynamic_broadcast_gradient_args.cc | 46 } else if (input_arg->isa<abstract::AbstractTuple>()) { in CheckInputsAndGetShape()
47 auto x_shape = dyn_cast<abstract::AbstractTuple>(input_arg); in CheckInputsAndGetShape()
|
| /third_party/mindspore/tests/ut/cpp/optimizer/ |
| D | clean_test.cc | 34 using mindspore::abstract::AbstractTuple;
59 AbstractBasePtr para_tuple = std::make_shared<AbstractTuple>(para_elem); in SetUp()
66 …AbstractBasePtr app_tuple_inner = std::make_shared<AbstractTuple>(AbstractBasePtrList{app_int, app… in SetUp()
67 …AbstractBasePtr app_tuple = std::make_shared<AbstractTuple>(AbstractBasePtrList{app_float, app_tup… in SetUp()
211 if (dyn_cast<AbstractTuple>(dt) != nullptr) { in TEST_F()
223 if (dyn_cast<AbstractTuple>(dt) != nullptr) { in TEST_F()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_structures.cc | 27 return std::make_shared<AbstractTuple>(args_spec_list); in InferImplMakeTuple()
40 AbstractTuplePtr keys = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplMakeDict()
41 AbstractTuplePtr values = CheckArg<AbstractTuple>(op_name, args_spec_list, 1); in InferImplMakeDict()
219 return InferTupleOrListGetItem<AbstractTuple>(primitive->name(), args_spec_list); in InferImplTupleGetItem()
229 return InferTupleOrListSetItem<AbstractTuple>(primitive->name(), args_spec_list); in InferImplTupleSetItem()
299 return std::make_shared<AbstractTuple>(keys); in InferImplDictGetKeys()
312 return std::make_shared<AbstractTuple>(values); in InferImplDictGetValues()
330 return InferTupleOrListOrDictLen<AbstractTuple>(primitive->name(), args_spec_list); in InferImplTupleLen()
|
| D | prim_nn.cc | 182 return std::make_shared<AbstractTuple>(rets); in InferImplBatchNorm()
200 return std::make_shared<AbstractTuple>(rets); in InferImplFusedSparseAdam()
430 return std::make_shared<AbstractTuple>(args_list); in InferImplBpropCut()
447 return std::make_shared<AbstractTuple>(ret); in InferImplDropout()
457 return std::make_shared<AbstractTuple>(elements); in InferImplSparseApplyFtrl()
468 return std::make_shared<AbstractTuple>(elements); in InferImplSparseApplyProximalAdagrad()
476 return std::make_shared<AbstractTuple>(elements); in InferImplSGD()
513 return std::make_shared<AbstractTuple>(elements); in InferImplCTCGreedyDecoder()
573 return std::make_shared<AbstractTuple>(elements); in InferImplComputeAccidentalHits()
|
| D | abstract_value.h | 393 class MS_CORE_API AbstractTuple : public AbstractSequeue {
395 explicit AbstractTuple(const AbstractBasePtrList &elements) : AbstractSequeue(elements) {} in AbstractTuple() function
397 ~AbstractTuple() override = default;
398 MS_DECLARE_PARENT(AbstractTuple, AbstractSequeue) in MS_DECLARE_PARENT() argument
404 …AbstractBasePtr Clone() const override { return std::make_shared<AbstractTuple>(ElementsClone()); } in Clone()
406 …AbstractBasePtr Broaden() const override { return std::make_shared<AbstractTuple>(ElementsBroaden(… in Broaden()
408 …AbstractBasePtr PartialBroaden() const override { return std::make_shared<AbstractTuple>(ElementsP… in PartialBroaden()
410 …AbstractBasePtr Join(const AbstractBasePtr &other) override { return ElementsJoin<AbstractTuple>(o… in Join()
414 bool operator==(const AbstractTuple &other) const;
420 using AbstractTuplePtr = std::shared_ptr<AbstractTuple>;
|
| D | prim_arrays.cc | 58 auto xs = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplBroadCastShape()
59 auto ys = CheckArg<AbstractTuple>(op_name, args_spec_list, 1); in InferImplBroadCastShape()
89 return std::make_shared<AbstractTuple>(elems); in InferImplBroadCastShape()
97 auto arg = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplStack()
176 return std::make_shared<AbstractTuple>(elements); in InferImplUnique()
207 AbstractTuplePtr dout = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplUniqueGrad()
463 return std::make_shared<AbstractTuple>(elements); in InferImplMapCacheIdx()
538 return std::make_shared<AbstractTuple>(elements); in InferImplSubAndFilter()
867 return std::make_shared<AbstractTuple>(output_list); in InferImplSplit()
938 if (args_spec_list[0]->isa<AbstractTuple>()) { in InferImplConcat()
[all …]
|
| D | prim_debug.cc | 37 return std::make_shared<AbstractTuple>(AbstractBasePtrList({tensor_value->Broaden()})); in InferImplDebug()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | value_test.cc | 29 using AbstractTuple = abstract::AbstractTuple; typedef
76 …AbstractBasePtr ta = std::make_shared<AbstractTuple>(AbstractBasePtrList({boola, i64a, f32a, sa, v… in TEST_F()
|
| /third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/ |
| D | data_test.cc | 53 AbstractTuple t1 = AbstractTuple(base_list); in TEST_F()
75 …AbstractBasePtr abs_func_tuple = std::make_shared<AbstractTuple>(AbstractBasePtrList({abs_f1, abs_… in TEST_F()
85 abs_func_tuple = std::make_shared<AbstractTuple>(AbstractBasePtrList({base1, abs_f2})); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | batch_norm_bert_fission.cc | 80 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce()
112 auto bn_abstract_tuple = dyn_cast<abstract::AbstractTuple>(bn->abstract()); in CreateBNTrainingUpdateV2()
121 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateV2()
|
| D | batch_norm_grad_infer_fission.cc | 85 auto bn_grad_abstract_tuple = dyn_cast<abstract::AbstractTuple>(bn_grad->abstract()); in CreateBNInferGrad()
131 auto bn_grad_abstract_tuple = dyn_cast<abstract::AbstractTuple>(bn_grad->abstract()); in CreateBNTrainingUpdateGrad()
139 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingUpdateGrad()
|
| D | single_batch_norm_fission.cc | 46 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in CreateBNTrainingReduce()
77 auto bn_abstract_tuple = dyn_cast<abstract::AbstractTuple>(bn->abstract()); in CreateBNTrainingUpdateV3()
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/common/ |
| D | restore_abs_input_in_backed_infer_test.cc | 44 EXPECT_EQ(args_spec_list[1]->isa<abstract::AbstractTuple>(), true); in InferImplAttrTest()
52 EXPECT_EQ(args_spec_list[1]->isa<abstract::AbstractTuple>(), true); in InferImplDynamicInputTest()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | validator.cc | 41 using mindspore::abstract::AbstractTuple;
118 …ptrBase->isa<AbstractType>() || ptrBase->isa<AbstractFunction>() || ptrBase->isa<AbstractTuple>() … in ValidateAbstract()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | unpack_call.cc | 39 using mindspore::abstract::AbstractTuple;
62 if (args_spec_list[index]->isa<AbstractTuple>()) { in GenerateFuncGraph()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | optimizer_unify_output.cc | 43 …fAlgo::HasNodeAttr("optim_output_passed", cnode_ptr) && abstract->isa<abstract::AbstractTuple>()) { in ProcessOutput()
52 auto abstract_tuple = std::make_shared<abstract::AbstractTuple>(abstract_list); in ProcessOutput()
|