Home
last modified time | relevance | path

Searched refs:AbstractTuplePtr (Results 1 – 25 of 47) sorted by relevance

12

/third_party/mindspore/tests/ut/cpp/operator/
Dcomposite_test.cc37 using AbstractTuplePtr = abstract::AbstractTuplePtr; typedef
162 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
188 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
214 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
240 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(tupleSliceGraphPtr, args_spec_list).in… in TEST_F()
259 AbstractTuplePtr tensor_tuple = std::make_shared<AbstractTuple>(eles); in TEST_F()
267 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(unPackCallGraphPtr, args_spec_list).in… in TEST_F()
286 AbstractTuplePtr tensor_tuple = std::make_shared<AbstractTuple>(eles); in TEST_F()
294 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(unPackCallGraphPtr, args_spec_list).in… in TEST_F()
316 …AbstractTuplePtr ret = dyn_cast<AbstractTuple>(engine_->Run(zip_op_graph, args_spec_list).inferred… in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dgraph_transform.cc61 const abstract::AbstractTuplePtr &abs) { in TransformTupleArgument()
68 …uto nodes = TransformTupleArgument(fg, elem_node, elements[i]->cast<abstract::AbstractTuplePtr>()); in TransformTupleArgument()
88 auto nodes = TransformTupleArgument(fg, cinputs[i], abs->cast<abstract::AbstractTuplePtr>()); in TransformCallGraph()
111 auto nodes = TransformTupleArgument(fg, cinputs[i], abs->cast<abstract::AbstractTuplePtr>()); in TransformPartial()
133 auto nodes = TransformTupleArgument(fg, cinputs[i], abs->cast<abstract::AbstractTuplePtr>()); in TransformSwitchCall()
Dgraph_transform.h33 const abstract::AbstractTuplePtr &abs);
51 …AnfNodePtr GenerateTupleParams(const abstract::AbstractTuplePtr &tuple_abs, const FuncGraphPtr &fg, in GenerateTupleParams()
58 inputs.push_back(GenerateTupleParams(item->cast<abstract::AbstractTuplePtr>(), fg, params)); in GenerateTupleParams()
80 auto tuple_abs = abs->cast<abstract::AbstractTuplePtr>(); in TransformGraphParam()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dzip_operation.h41 using AbstractTuplePtr = abstract::AbstractTuplePtr; variable
Dunpack_call.cc40 using mindspore::abstract::AbstractTuplePtr;
63 auto arg_tuple = args_spec_list[index]->cast<AbstractTuplePtr>(); in GenerateFuncGraph()
Dcomposite.cc397 bool CheckSequenceAllTensor(const abstract::AbstractTuplePtr &tuple) { in CheckSequenceAllTensor()
401 CheckSequenceAllTensor((*tuple)[i]->cast<abstract::AbstractTuplePtr>()))) { in CheckSequenceAllTensor()
414 CheckSequenceAllTensor((*sequeue)[1]->cast<abstract::AbstractTuplePtr>()))); in CheckTailGradFristSequence()
866 abstract::AbstractTuplePtr a_tuple = dyn_cast<AbstractTuple>(abs_a); in GenerateFuncGraph()
867 abstract::AbstractTuplePtr b_tuple = dyn_cast<AbstractTuple>(abs_b); in GenerateFuncGraph()
933 void GenerateTupleSliceParameter(const AbstractTuplePtr &tuple, const AbstractSlicePtr &slice, int6… in GenerateTupleSliceParameter()
982 AbstractTuplePtr tuple = abstract::CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in GenerateFuncGraph()
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_tuplegetitem_fusion.cc63 if (abstr != nullptr && utils::isa<abstract::AbstractTuplePtr>(abstr)) { in Process()
64 auto elements = utils::cast<abstract::AbstractTuplePtr>(abstr)->elements(); in Process()
/third_party/mindspore/mindspore/core/ops/
Daddn.cc34 ? input_args[0]->cast<abstract::AbstractTuplePtr>()->elements() in AddNInferShape()
72 ? input_args[0]->cast<abstract::AbstractTuplePtr>()->elements() in AddNInferType()
Dstrided_slice.cc121 auto temp_strides = stride_arg->cast<abstract::AbstractTuplePtr>()->BuildValue(); in CheckAndGetValidStrides()
200 auto tuple_begin_v = input_args[kInputIndex1]->cast<abstract::AbstractTuplePtr>(); in StridedSliceInferShape()
206 auto tuple_end_v = input_args[kInputIndex2]->cast<abstract::AbstractTuplePtr>(); in StridedSliceInferShape()
Dconcat.cc40 auto input_tuple = input_args[0]->cast<abstract::AbstractTuplePtr>(); in ConcatInfer()
Dreshape.cc41 auto shape = input_args[1]->cast<abstract::AbstractTuplePtr>(); in ReshapeInfer()
Dtile.cc64 auto get_cast_temp = input_args[1]->cast<abstract::AbstractTuplePtr>(); in TileInferShape()
Dneighborexchange.cc113 auto abstract_tuple = input_args[0]->cast<abstract::AbstractTuplePtr>(); in Check()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dbranch_culling.cc447 …abstract::AbstractTuplePtr true_branch_tuple = true_branch_abs->cast<abstract::AbstractTuplePtr>(); in GraphOutputCompatible()
448 …abstract::AbstractTuplePtr false_branch_tuple = false_branch_abs->cast<abstract::AbstractTuplePtr>… in GraphOutputCompatible()
492 auto true_branch_tuple = branch_output_abs[0]->cast<abstract::AbstractTuplePtr>(); in GenerateMergeNodes()
493 auto false_branch_tuple = branch_output_abs[1]->cast<abstract::AbstractTuplePtr>(); in GenerateMergeNodes()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.cc215 AbstractBasePtr DoInferReduceShape(const AbstractTuplePtr &x_shape, const ValuePtr &x_shp_value, in DoInferReduceShape()
340 AbstractTuplePtr input = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplTupleReversed()
355 AbstractTuplePtr shape_x = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplReduceShape()
395 AbstractTuplePtr shape_x = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplTupleDiv()
396 AbstractTuplePtr div_shp = CheckArg<AbstractTuple>(op_name, args_spec_list, 1); in InferImplTupleDiv()
445 AbstractTuplePtr input = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplTuple2Array()
462 AbstractTuplePtr shape_x = CheckArg<AbstractTuple>(op_name, args_spec_list, 0); in InferImplShapeMul()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc70 …if (!mindspore::utils::isa<mindspore::abstract::AbstractTuplePtr>(get_item_input_cnode->abstract()… in GetShapeVectorFromCNode()
75 … mindspore::utils::cast<mindspore::abstract::AbstractTuplePtr>(get_item_input_cnode->abstract()); in GetShapeVectorFromCNode()
/third_party/mindspore/mindspore/core/ops/grad/
Dflatten_grad.cc32 auto input_shape = input_args[1]->cast<abstract::AbstractTuplePtr>(); in FlattenGradInfer()
/third_party/mindspore/mindspore/core/abstract/
Dabstract_value.h420 using AbstractTuplePtr = std::shared_ptr<AbstractTuple>; variable
716 const AbstractTuplePtr dense_shape() const { return dense_shape_; } in dense_shape()
717 void set_dense_shape(const AbstractTuplePtr &dense_shape) { dense_shape_ = dense_shape; } in set_dense_shape()
728 AbstractTuplePtr dense_shape_;
745 const AbstractTuplePtr dense_shape() const { return dense_shape_; } in dense_shape()
746 void set_dense_shape(const AbstractTuplePtr &dense_shape) { dense_shape_ = dense_shape; } in set_dense_shape()
757 AbstractTuplePtr dense_shape_;
Dabstract_value.cc1274 clone->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in Clone()
1296 broaden->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in Broaden()
1319 broaden->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in BroadenWithShape()
1366 clone->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in Clone()
1388 broaden->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in Broaden()
1413 broaden->set_dense_shape(dense_clone->cast<AbstractTuplePtr>()); in BroadenWithShape()
Dprim_structures.cc40 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()
Dprim_statement.cc86 AbstractTuplePtr branches_abs = CheckArg<AbstractTuple>(op_name, args_spec_list, 1); in InferImplSwitchLayer()
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dprim_test.cc39 using AbstractTuplePtr = std::shared_ptr<AbstractTuple>; typedef in mindspore::abstract::UTPrimUtils
64 static AbstractTuplePtr ShapeOf(std::initializer_list<int64_t> vals) { in ShapeOf()
285 AbstractTuplePtr res_J = dyn_cast<AbstractTuple>(res); in TEST_F()
344 …AbstractTuplePtr res = dyn_cast<AbstractTuple>(engine_->Run(func_graph, args_spec_list).inferred->… in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/common/
Drestore_abs_input_in_backed_infer_test.cc53 auto item = args_spec_list[1]->cast<abstract::AbstractTuplePtr>(); in InferImplDynamicInputTest()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Drewrite_output_shape.cc54 auto node_abstracts = node->abstract()->cast<abstract::AbstractTuplePtr>()->elements(); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_cast.cc57 auto abs = cnode->abstract()->cast<abstract::AbstractTuplePtr>(); in InsertCastForMultipleOutput()

12