Home
last modified time | relevance | path

Searched refs:AbstractTensorPtr (Results 1 – 25 of 55) sorted by relevance

123

/third_party/mindspore/tests/ut/cpp/operator/
Dcomposite_test.cc40 using AbstractTensorPtr = abstract::AbstractTensorPtr; typedef
67 static AbstractTensorPtr ArrayInt32Of(std::initializer_list<int64_t> shp) { in ArrayInt32Of()
254 AbstractTensorPtr tensor = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
260 AbstractTensorPtr arr_x = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
261 AbstractTensorPtr arr_y = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
262 AbstractTensorPtr arr_z = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
281 AbstractTensorPtr tensor = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
287 AbstractTensorPtr arr_x = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
288 AbstractTensorPtr arr_y = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
289 AbstractTensorPtr arr_z = UTCompositeUtils::ArrayInt32Of({2, 3, 4}); in TEST_F()
/third_party/mindspore/mindspore/core/abstract/
Dparam_validator.h35 TypePtr CheckTensorDType(const AbstractTensorPtr &tensor, const TypePtrList &accepts,
44 …ePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractTen…
46 …ePtr CheckDtypeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractTen…
Dabstract_value.h359 using AbstractTensorPtr = std::shared_ptr<AbstractTensor>; variable
360 using AbstractTensorPtrList = std::vector<AbstractTensorPtr>;
651 AbstractRef(const AbstractBasePtr &ref_key, const AbstractTensorPtr &ref_value);
659 auto abs_tensor = AbstractTensor::Clone()->cast<AbstractTensorPtr>(); in Clone()
667 inline AbstractTensorPtr ref() { return shared_from_base<AbstractTensor>(); } in ref()
672 auto abs_tensor = AbstractTensor::Broaden()->cast<AbstractTensorPtr>(); in Broaden()
712 const AbstractTensorPtr indices() const { return indices_; } in MS_DECLARE_PARENT()
713 void set_indices(const AbstractTensorPtr &indices) { indices_ = indices; } in set_indices()
714 const AbstractTensorPtr values() const { return values_; } in values()
715 void set_values(const AbstractTensorPtr &values) { values_ = values; } in set_values()
[all …]
Dprim_arrays.cc103 AbstractTensorPtr tensor_base = CheckArg<AbstractTensor>(op_name, arg->elements(), 0); in InferImplStack()
115 AbstractTensorPtr tensor = CheckArg<AbstractTensor>(op_name, arg->elements(), i); in InferImplStack()
124 AbstractTensorPtr ret = dyn_cast<AbstractTensor>(tensor_base->Broaden()); in InferImplStack()
139 AbstractTensorPtr input = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplUnique()
185 AbstractTensorPtr input = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplPadAndShift()
497 AbstractTensorPtr ret = in InferImplCacheSwapTable()
510 …AbstractTensorPtr ret = std::make_shared<AbstractTensor>(input_x->element(), std::make_shared<Shap… in InferImplUpdateCache()
584 AbstractTensorPtr params = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplGatherV2()
585 AbstractTensorPtr indices = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplGatherV2()
592 auto axis = args_spec_list[aixs_index]->cast<AbstractTensorPtr>(); in InferImplGatherV2()
[all …]
Dparam_validator.cc52 TypePtr CheckTensorDType(const AbstractTensorPtr &tensor, const TypePtrList &accepts, in CheckTensorDType()
113 …ePtr CheckShapeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractTen… in CheckShapeSame()
127 …ePtr CheckDtypeSame(const std::string &op, const AbstractTensorPtr &tensor_base, const AbstractTen… in CheckDtypeSame()
Dprim_nn.cc46 AbstractTensorPtr input_tensor = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplPooling()
133 AbstractTensorPtr input_x = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplBatchNorm()
143 AbstractTensorPtrList tensorPtrList = std::vector<AbstractTensorPtr>(); in InferImplBatchNorm()
160 AbstractTensorPtr arg_spec = CheckArg<AbstractTensor>(op_name, args_spec_list, i); in InferImplBatchNorm()
172 AbstractTensorPtr input_gamma = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplBatchNorm()
178 AbstractTensorPtr output = std::make_shared<AbstractTensor>(input_x->element(), output_shape_ptr); in InferImplBatchNorm()
180 …AbstractTensorPtr output_gamma = std::make_shared<AbstractTensor>(input_gamma->element(), gamma_sh… in InferImplBatchNorm()
234 void CheckShape(const std::string &op_name, const ShapeVector &w_shape, const AbstractTensorPtr &in… in CheckShape()
248 AbstractTensorPtr input_x = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplConv2D()
258 AbstractTensorPtr input_w = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplConv2D()
[all …]
Dutils.h58 …tr GetBroadcastShape(const std::string &op, const AbstractTensorPtr &tensor_x, const AbstractTenso…
Dabstract_value.cc949 AbstractRef::AbstractRef(const AbstractBasePtr &ref_key, const AbstractTensorPtr &ref_value) in AbstractRef()
1000 return join_abs->cast<AbstractTensorPtr>(); in Join()
1012 auto ref_tensor = ref->cast<AbstractTensorPtr>(); in Join()
1272 clone->set_indices(indices_clone->cast<AbstractTensorPtr>()); in Clone()
1273 clone->set_values(value_clone->cast<AbstractTensorPtr>()); in Clone()
1294 broaden->set_indices(indices_clone->cast<AbstractTensorPtr>()); in Broaden()
1295 broaden->set_values(value_clone->cast<AbstractTensorPtr>()); in Broaden()
1317 broaden->set_indices(indices_clone->cast<AbstractTensorPtr>()); in BroadenWithShape()
1318 broaden->set_values(value_clone->cast<AbstractTensorPtr>()); in BroadenWithShape()
1364 clone->set_indices(indices_clone->cast<AbstractTensorPtr>()); in Clone()
[all …]
Dprim_others.cc60 auto dflt_tensor = dflt->cast<AbstractTensorPtr>(); in InferImplEnvGetItem()
122 auto tensor = args_spec_list[1]->cast<abstract::AbstractTensorPtr>(); in InferImplMakeRef()
432 AbstractTensorPtr ret = std::make_shared<AbstractTensor>( in InferImplAllSwap()
558 AbstractTensorPtr input = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplGpuConvertToDynamicShape()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_property_checker.cc28 using AbstractTensorPtr = mindspore::abstract::AbstractTensorPtr; typedef
64 input_dims = SizeToInt(input_abstract->cast<AbstractTensorPtr>()->shape()->shape().size()); in CheckStridedSlice()
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc86 if (!mindspore::utils::isa<mindspore::abstract::AbstractTensorPtr>(cnode_abstract)) { in GetShapeVectorFromCNode()
90 auto cnode_abstract_tensor = cnode_abstract->cast<mindspore::abstract::AbstractTensorPtr>(); in GetShapeVectorFromCNode()
110 if (utils::isa<abstract::AbstractTensorPtr>(cnode->abstract())) { in GetTypeFromNode()
111 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in GetTypeFromNode()
/third_party/mindspore/mindspore/core/ops/
Dgather.cc34 abstract::AbstractTensorPtr params = in GatherInfer()
36 abstract::AbstractTensorPtr indices = in GatherInfer()
52 auto axis = input_args[kInputIndex2]->cast<abstract::AbstractTensorPtr>(); in GatherInfer()
Dembedding_lookup.cc45 auto params = input_args[0]->cast<abstract::AbstractTensorPtr>(); in EmbeddingLookupInfer()
47 auto indices = input_args[1]->cast<abstract::AbstractTensorPtr>(); in EmbeddingLookupInfer()
Ddepth_to_space.cc54 auto input_x = input_args[0]->cast<abstract::AbstractTensorPtr>(); in DepthToSpaceInfer()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dunused_cast_node_remove_pass.cc47 if (!utils::isa<abstract::AbstractTensorPtr>(abstract_base)) { in Run()
52 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in Run()
Dnode_infershape.cc395 if (!utils::isa<abstract::AbstractTensorPtr>(elements[i])) { in GetCNodeOutputTensors()
400 if (utils::isa<abstract::AbstractTensorPtr>(elements[i])) { in GetCNodeOutputTensors()
401 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(elements[i]); in GetCNodeOutputTensors()
416 if (utils::isa<abstract::AbstractTensorPtr>(cnode->abstract())) { in GetCNodeOutputTensors()
417 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(cnode->abstract()); in GetCNodeOutputTensors()
Dadd_tensor_array.cc144 if (!utils::isa<abstract::AbstractTensorPtr>(abstract)) { in Process()
150 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dconv2d_info.cc106 if (!utils::isa<abstract::AbstractTensorPtr>(input_node_abstract)) { in CheckIfSplit()
110 if (!utils::isa<abstract::AbstractTensorPtr>(weight_node_abstract)) { in CheckIfSplit()
114 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(input_node_abstract); in CheckIfSplit()
118 abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(weight_node_abstract); in CheckIfSplit()
/third_party/mindspore/mindspore/core/ops/grad/
Dflatten_grad.cc30 auto input_x = input_args[0]->cast<abstract::AbstractTensorPtr>(); in FlattenGradInfer()
/third_party/mindspore/mindspore/lite/tools/anf_exporter/
Dfetch_content.cc97 if (!utils::isa<abstract::AbstractTensorPtr>(abstract_base)) { in GetDataTypeAndShape()
101 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract_base); in GetDataTypeAndShape()
119 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(valueAbstract); in FetchFromTensorValue()
377 if (!utils::isa<abstract::AbstractTensorPtr>(abstract)) { in FetchDataFromCNode()
381 auto abstract_tensor = utils::cast<abstract::AbstractTensorPtr>(abstract); in FetchDataFromCNode()
/third_party/mindspore/mindspore/lite/test/common/
Dimport_from_meta_graphT.h45 …static abstract::AbstractTensorPtr ConvertTensorToAbstractTensor(const std::unique_ptr<schema::Ten…
/third_party/mindspore/mindspore/core/ir/
Dpattern_matcher.h646 auto tensor_abstract = node->abstract()->cast<abstract::AbstractTensorPtr>();
667 auto x_abstract = x->abstract()->cast<abstract::AbstractTensorPtr>();
717 auto tensor_abstract = node->abstract()->cast<abstract::AbstractTensorPtr>();
770 auto tensor_abstract = node->abstract()->cast<abstract::AbstractTensorPtr>();
964 auto tensor_1_abstract = vnode_1->abstract()->cast<abstract::AbstractTensorPtr>();
965 auto tensor_2_abstract = vnode_2->abstract()->cast<abstract::AbstractTensorPtr>();
977 auto tensor_3_abstract = node_3->abstract()->cast<abstract::AbstractTensorPtr>();
/third_party/mindspore/tests/ut/cpp/pipeline/static_analysis/
Dprim_test.cc38 using AbstractTensorPtr = std::shared_ptr<AbstractTensor>; typedef in mindspore::abstract::UTPrimUtils
49 static AbstractTensorPtr ArrayFloat64Of(std::initializer_list<int64_t> shp) { in ArrayFloat64Of()
54 static AbstractTensorPtr ArrayFloat32Of(std::initializer_list<int64_t> shp) { in ArrayFloat32Of()
59 static AbstractTensorPtr ArrayInt32Of(std::initializer_list<int64_t> shp) { in ArrayInt32Of()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Darithmetic_simplify.cc132 auto x_abstract = x_->abstract()->cast<abstract::AbstractTensorPtr>(); in operator ()()
146 auto z_abstract = z_->abstract()->cast<abstract::AbstractTensorPtr>(); in operator ()()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dcomposite.h45 using AbstractTensorPtr = abstract::AbstractTensorPtr; variable

123