Home
last modified time | relevance | path

Searched defs:shape_vec (Results 1 – 25 of 73) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dclip_by_norm_fission.cc89 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSquareNode() argument
98 … const AnfNodePtr &clip_by_norm, const ShapeVector &shape_vec, in CreateReduceSumNode()
147 … const ShapeVector &shape_vec, const TypeId &type_id, in CreateConstantNode()
175 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSelectNode() argument
205 … const AnfNodePtr &clip_norm, const ShapeVector &shape_vec, in CreateMulNode()
221 const AnfNodePtr &divisor, const ShapeVector &shape_vec, in CreateDivNode()
231 … const ShapeVector &shape_vec, const TypeId &src_type_id, in CreateCastNode()
272 auto shape_vec = GetOutputInferShape(clip_by_norm); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dascend_clip_by_norm_fission.cc88 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSquareNode() argument
97 … const AnfNodePtr &clip_by_norm, const ShapeVector &shape_vec, in CreateReduceSumNode()
143 … const ShapeVector &shape_vec, const TypeId &type_id, in CreateConstantNode()
171 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSelectNode() argument
201 … const AnfNodePtr &clip_norm, const ShapeVector &shape_vec, in CreateMulNode()
217 … const AnfNodePtr &divisor, const ShapeVector &shape_vec, in CreateDivNode()
227 … const ShapeVector &shape_vec, const TypeId &src_type_id, in CreateCastNode()
271 auto shape_vec = GetOutputInferShape(clip_by_norm); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dutils.cc119 ShapeVector shape_vec(shape, shape + out_dims[0]); in BuildShape() local
125 ShapeVector shape_vec(shape, shape + out_dims[i]); in BuildShape() local
161 auto shape_vec = shape->shape(); in BuildAbstract() local
186 ShapeVector shape_vec{}; in GetAbstract() local
196 ShapeVector shape_vec{}; in GetAbstract() local
199 ShapeVector shape_vec(shape, shape + shape_size); in GetAbstract() local
Dtensor.cc50 ShapeVector shape_vec(shape, shape + shape_size); in MSNewTensor() local
67 ShapeVector shape_vec(shape, shape + shape_size); in MSNewTensorFromFile() local
104 ShapeVector shape_vec(shape, shape + shape_size); in MSNewTensorWithSrcType() local
273 ShapeVector shape_vec(shape, shape + dim); in MSTensorSetShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dclip_by_norm_fission.cc139 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSquareNode() argument
148 … const AnfNodePtr &clip_by_norm, const ShapeVector &shape_vec, in CreateReduceSumNode()
205 … const ShapeVector &shape_vec, const TypeId &type_id, in CreateConstantNode()
231 … const ShapeVector &shape_vec, const TypeId &type_id) const { in CreateSelectNode() argument
266 … const AnfNodePtr &clip_norm, const ShapeVector &shape_vec, in CreateMulNode()
283 const AnfNodePtr &divisor, const ShapeVector &shape_vec, in CreateDivNode()
294 … const ShapeVector &shape_vec, const TypeId &src_type_id, in CreateCastNode()
346 auto shape_vec = GetOutputInferShape(clip_by_norm); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dnon_zero_with_value.cc56 auto shape_vec = x_shape->GetShapeVector(); in NonZeroWithValueInfer() local
87 auto shape_vec = x_shape->GetShapeVector(); in InferShape() local
Dmake_csrtensor.cc73 auto shape_vec = GetShapeValue(primitive, shape); in MakeCSRTensorInfer() local
Dmap_tensor_get_keys.cc39 ShapeVector shape_vec = {abstract::Shape::kShapeDimAny}; in MapTensorGetKeysInferShape() local
Dmap_tensor_get_values.cc45 ShapeVector shape_vec = {abstract::Shape::kShapeDimAny}; in MapTensorGetValuesInferShape() local
Dmap_tensor_get.cc48 auto shape_vec = key_tensor_shape->shape(); in MapTensorGetInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_frontend_func_impl/
Dones_like.cc31 auto shape_vec = shape->shape(); in InferValue() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dsqueeze_axis.cc37 auto shape_vec = common::AnfAlgo::GetOutputInferShape(squeeze_cnode->input(kIndex1), 0); in SqueezeAxis() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dargmax_fusion_mapper.cc51 std::vector<int64_t> shape_vec = {}; in Mapper() local
Dtopk_fusion_mapper.cc87 std::vector<int64_t> shape_vec = {}; in Mapper() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_utils.cc275 inline bool IsSameRank(const Shape &shape_vec, const Shape &targe_shape_vec) { in IsSameRank()
279 bool HasAssebledDynamicDim(const Shape &shape_vec, const AssembledDynamicDimsMapping &dyn_dims_mapp… in HasAssebledDynamicDim()
301 void MatchingAccordingToPrime(const Shape &shape_vec, const AssembledDynamicDimsMapping &dyn_dims_m… in MatchingAccordingToPrime()
352 void MatchingAccordingToIndex(const Shape &shape_vec, const AssembledDynamicDimsMapping &dyn_dims_m… in MatchingAccordingToIndex()
432 std::vector<int64_t> shape_vec = GetValue<std::vector<int64_t>>(param.first.second); in ConvertConstParamToDynamic() local
532 Shape shape_vec = GetValue<Shape>(param.first.second); in ConvertStridedSliceInputs() local
548 bool WhetherMatchingIsNeededForReshape(const Shape &shape_vec, const TensorRedistributionPtr &tenso… in WhetherMatchingIsNeededForReshape()
566 inline bool HasOnlyOneDynamicAxis(const Shape &shape_vec, in HasOnlyOneDynamicAxis()
576 …icAxisToNegOne(const TensorRedistributionPtr &tensor_redistribution_from_cnode, Shape *shape_vec) { in ReplaceDynamicAxisToNegOne()
607 Shape shape_vec = GetValue<Shape>(shape_param.first.second); in ConvertReshapeInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dnormal_float_float.cc59 std::vector<int64_t> shape_vec = shape_array.ToVector(); in InferShape() local
Ddropout_ext.cc25 int64_t CalMaskShape(const PrimitivePtr &primitive, const ShapeVector &shape_vec) { in CalMaskShape()
Dreshape.cc47 std::vector<int64_t> shape_vec = shape_array.ToVector(); in InferShape() local
Dmatmul_fusion_utils.cc29 std::vector<int64_t> shape_vec = shape_array.ToVector(); in InferenceMultiMatmulInferShape() local
Daddn.cc82 ShapeVector shape_vec; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dsplit_shared_bias.cc46 ShapeVector shape_vec(data_info.shape_.begin(), data_info.shape_.end()); in CloneParameterNode() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/core/utils/
Dcheck_convert_utils_test.cc28 AbstractBasePtr MakeTensorAbstract(const ShapeVector &shape_vec, const TypePtr &elem_type) { in MakeTensorAbstract()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_cmp_utils.h43 std::vector<abstract::BaseShapePtr> shape_vec; in MakeOutputTupleShapeAndType() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dcore_infershape_pass.cc275 ShapeVector shape_vec = {-1}; in SetSubGraphInput() local
295 ShapeVector shape_vec(data_info.shape_.begin(), data_info.shape_.end()); in SetSubGraphInput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Dop_util.cc45 std::vector<int64_t> shape_vec(shape_size, 0); in ToVector() local

123