| /third_party/mindspore/mindspore/lite/mindir/src/ |
| D | mindir_nnrt_lite_graph_to_model.cc | 177 std::vector<int8_t> ConvertActivation(PrimitivePtr primitive) { in ConvertActivation()
201 std::vector<int8_t> ConvertAddFusion(PrimitivePtr primitive) { in ConvertAddFusion()
220 std::vector<int8_t> ConvertArgMaxFusion(PrimitivePtr primitive) { in ConvertArgMaxFusion()
242 std::vector<int8_t> ConvertAvgPoolFusion(PrimitivePtr primitive) { in ConvertAvgPoolFusion()
289 std::vector<int8_t> ConvertBatchToSpaceND(PrimitivePtr primitive) { in ConvertBatchToSpaceND()
328 std::vector<int8_t> ConvertBiasAdd(PrimitivePtr primitive) { in ConvertBiasAdd()
346 std::vector<int8_t> ConvertCast(PrimitivePtr primitive) { in ConvertCast()
364 std::vector<int8_t> ConvertConcat(PrimitivePtr primitive) { in ConvertConcat()
383 std::vector<int8_t> ConvertConv2DFusion(PrimitivePtr primitive) { in ConvertConv2DFusion()
438 std::vector<int8_t> ConvertConv2dTransposeFusion(PrimitivePtr primitive) { in ConvertConv2dTransposeFusion()
[all …]
|
| D | mindir.cc | 37 ActivationType MindIR_Activation_GetActivationType(ConstPrimitivePtr primitive) { in MindIR_Activation_GetActivationType()
53 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_Activation_SetActivationType()
71 float MindIR_Activation_GetAlpha(ConstPrimitivePtr primitive) { in MindIR_Activation_GetAlpha()
85 void MindIR_Activation_SetAlpha(PrimitivePtr *primitive, float alpha) { in MindIR_Activation_SetAlpha()
102 float MindIR_Activation_GetMinVal(ConstPrimitivePtr primitive) { in MindIR_Activation_GetMinVal()
116 void MindIR_Activation_SetMinVal(PrimitivePtr *primitive, float min_val) { in MindIR_Activation_SetMinVal()
133 float MindIR_Activation_GetMaxVal(ConstPrimitivePtr primitive) { in MindIR_Activation_GetMaxVal()
147 void MindIR_Activation_SetMaxVal(PrimitivePtr *primitive, float max_val) { in MindIR_Activation_SetMaxVal()
164 bool MindIR_Activation_GetApproximate(ConstPrimitivePtr primitive) { in MindIR_Activation_GetApproximate()
178 void MindIR_Activation_SetApproximate(PrimitivePtr *primitive, bool approximate) { in MindIR_Activation_SetApproximate()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/train/ |
| D | train_populate_parameter_v0.cc | 37 OpParameter *DefaultPopulateParameter(const void *primitive) { in DefaultPopulateParameter()
94 OpParameter *PopulateSmoothL1LossParameter(const void *primitive) { in PopulateSmoothL1LossParameter()
114 OpParameter *PopulateSmoothL1LossGradParameter(const void *primitive) { in PopulateSmoothL1LossGradParameter()
134 OpParameter *PopulateApplyMomentumParameter(const void *primitive) { in PopulateApplyMomentumParameter()
156 OpParameter *PopulateBCEParameter(const void *primitive) { in PopulateBCEParameter()
169 OpParameter *PopulateBCEGradParameter(const void *primitive) { in PopulateBCEGradParameter()
182 OpParameter *PopulateAdamParameter(const void *primitive) { in PopulateAdamParameter()
202 OpParameter *PopulateSgdParameter(const void *primitive) { in PopulateSgdParameter()
225 OpParameter *PopulateSparseSoftmaxCrossEntropyParameter(const void *primitive) { in PopulateSparseSoftmaxCrossEntropyParameter()
246 OpParameter *PopulateSoftmaxCrossEntropyParameter(const void *primitive) { in PopulateSoftmaxCrossEntropyParameter()
[all …]
|
| D | train_populate_parameter.cc | 49 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSmoothL1LossParameter() local
64 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSmoothL1LossGradParameter() local
79 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateApplyMomentumParameter() local
93 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateBCEParameter() local
106 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateBCEGradParameter() local
120 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateAdamParameter() local
135 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSgdParameter() local
154 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSparseSoftmaxCrossEntropyWithLogitsParameter() local
170 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSoftmaxCrossEntropyParameter() local
183 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateMaxPoolGradParameter() local
[all …]
|
| /third_party/mindspore/mindspore/lite/src/common/ |
| D | prim_util.cc | 28 int GetPrimitiveType(const void *primitive, int schema_version) { in GetPrimitiveType()
40 const char *GetPrimitiveTypeName(const void *primitive, int schema_version) { in GetPrimitiveTypeName()
58 bool IsPartialNode(const void *primitive, int schema_version) { in IsPartialNode()
72 bool IsCallNode(const void *primitive, int schema_version) { in IsCallNode()
80 bool IsSwitchNode(const void *primitive, int schema_version) { in IsSwitchNode()
88 bool IsCustomNode(const void *primitive, int schema_version) { in IsCustomNode()
96 int GetPartialGraphIndex(const void *primitive, int schema_version) { in GetPartialGraphIndex()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 36 AbstractBasePtr InferImplIdentity(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplIdentity()
43 AbstractBasePtr InferImplEnvGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvGetItem()
76 AbstractBasePtr InferImplEnvSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvSetItem()
94 AbstractBasePtr InferImplEnvAdd(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvAdd()
157 AbstractBasePtr InferImplStateSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplStateSetItem()
170 AbstractBasePtr InferImplDepend(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplDepend()
193 AbstractBasePtr InferImplUpdateState(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUpdateState()
202 AbstractBasePtr InferImplMakeRowTensor(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeRowTensor()
265 …stractBasePtr InferImplRowTensorGetValues(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplRowTensorGetValues()
275 …tractBasePtr InferImplRowTensorGetIndices(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplRowTensorGetIndices()
[all …]
|
| D | prim_structures.cc | 35 AbstractBasePtr InferImplMakeDict(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeDict()
65 AbstractBasePtr InferImplMakeKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeKwarg()
81 AbstractBasePtr InferImplExtractKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplExtractKwarg()
103 AbstractBasePtr InferImplMakeSlice(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeSlice()
217 AbstractBasePtr InferImplTupleGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleGetItem()
222 AbstractBasePtr InferImplListGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplListGetItem()
227 AbstractBasePtr InferImplTupleSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleSetItem()
232 AbstractBasePtr InferImplListSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplListSetItem()
237 AbstractBasePtr InferImplDictGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplDictGetItem()
260 AbstractBasePtr InferImplDictSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplDictSetItem()
[all …]
|
| D | prim_arrays.cc | 29 AbstractBasePtr InferImplScalarToArray(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplScalarToArray()
38 AbstractBasePtr InferImplArrayToScalar(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplArrayToScalar()
52 AbstractBasePtr InferImplBroadCastShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBroadCastShape()
92 AbstractBasePtr InferImplStack(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplStack()
134 AbstractBasePtr InferImplUnique(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUnique()
179 AbstractBasePtr InferImplPadAndShift(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplPadAndShift()
201 AbstractBasePtr InferImplUniqueGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUniqueGrad()
226 …stractBasePtr InferImplUnsortedSegmentSum(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUnsortedSegmentSum()
282 …stractBasePtr InferImplUnsortedSegmentMax(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUnsortedSegmentMax()
333 …stractBasePtr InferImplUnsortedSegmentMin(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplUnsortedSegmentMin()
[all …]
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 62 const PrimitivePtr &primitive) { in SetAbstract()
89 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoConvQuantize() local
149 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoMulQuantize() local
194 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoLstmQuantize() local
225 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoGatherQuantize() local
269 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoOptimizerQuantize() local
317 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoMarkWeightQuantizeIfQuantized() local
345 …TUS WeightQuantizer::ProcessLstmWeightByIndex(const CNodePtr &cnode, const PrimitivePtr &primitive, in ProcessLstmWeightByIndex()
396 auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0)); in MarkWeightQuantizationInNodes() local
415 auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0)); in DoQuantize() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | hshrink.cc | 29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
35 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
44 AbstractBasePtr HShrinkInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in HShrinkInfer()
|
| D | assign_add.cc | 26 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
32 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
41 AbstractBasePtr AssignAddInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in AssignAddInfer()
|
| D | ones_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
47 AbstractBasePtr OnesLikeInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in OnesLikeInfer()
|
| D | diag.cc | 27 abstract::ShapePtr DiagInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr>… in DiagInferShape()
37 TypePtr PartInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args… in PartInferType()
43 AbstractBasePtr DiagInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in DiagInfer()
|
| D | zeros_like.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
45 AbstractBasePtr ZerosLikeInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in ZerosLikeInfer()
|
| D | rfft.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
34 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
49 AbstractBasePtr RfftInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in RfftInfer()
|
| D | soft_margin_loss.cc | 27 abstract::ShapePtr SoftMarginLossInferShape(const PrimitivePtr &primitive, in SoftMarginLossInferShape()
44 TypePtr SoftMarginLossInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &… in SoftMarginLossInferType()
56 …ractBasePtr SoftMarginLossInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in SoftMarginLossInfer()
|
| D | diag_part.cc | 29 abstract::ShapePtr DiagPartInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBase… in DiagPartInferShape()
47 TypePtr DiagPartInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_… in DiagPartInferType()
53 AbstractBasePtr DiagPartInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in DiagPartInfer()
|
| D | constant_of_shape.cc | 25 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
33 TypePtr InferType(const PrimitivePtr &primitive) { in InferType()
58 …actBasePtr ConstantOfShapeInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in ConstantOfShapeInfer()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | prim.h | 35 StandardPrimEvaluator(const PrimitivePtr &primitive, const StandardPrimitiveImplReg &eval_impl) in StandardPrimEvaluator()
56 explicit PythonPrimEvaluator(const PrimitivePyPtr primitive) in PythonPrimEvaluator()
71 …explicit DoSignatureEvaluator(const PrimitivePtr primitive) : Evaluator("DoSignatureEvaluator"), p… in DoSignatureEvaluator()
86 …explicit UnpackGraphEvaluator(const PrimitivePtr primitive) : Evaluator("UnpackGraphEvaluator"), p… in UnpackGraphEvaluator()
101 explicit MixedPrecisionCastEvaluator(const PrimitivePtr primitive) in MixedPrecisionCastEvaluator()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 149 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomCount() local
190 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomOperationType() local
215 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomRootId() local
229 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomSrcRank() local
243 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomDestRank() local
257 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomReceiveType() local
269 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomGroup() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/ |
| D | ops_front_infer_function.cc | 182 AbstractBasePtr InferImplHasType(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplHasType()
216 … const ValueSequeuePtr &axis_value_ptr, const PrimitivePtr &primitive) { in DoInferReduceShape()
250 …actBasePtr InferImplBroadcastGradientArgs(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplBroadcastGradientArgs()
286 AbstractBasePtr InferImplListMap(const AnalysisEnginePtr &engine, const PrimitivePtr &primitive, in InferImplListMap()
314 AbstractBasePtr InferImplListReduce(const AnalysisEnginePtr &engine, const PrimitivePtr &primitive, in InferImplListReduce()
335 AbstractBasePtr InferImplTupleReversed(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleReversed()
349 AbstractBasePtr InferImplReduceShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplReduceShape()
389 AbstractBasePtr InferImplTupleDiv(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleDiv()
440 AbstractBasePtr InferImplTuple2Array(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTuple2Array()
456 AbstractBasePtr InferImplShapeMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplShapeMul()
[all …]
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | soft_margin_loss_grad.cc | 27 abstract::ShapePtr SoftMarginLossGradInferShape(const PrimitivePtr &primitive, in SoftMarginLossGradInferShape()
42 TypePtr SoftMarginLossGradInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePt… in SoftMarginLossGradInferType()
55 …BasePtr SoftMarginLossGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in SoftMarginLossGradInfer()
|
| D | avg_pool_3d_grad.cc | 30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape()
49 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType()
63 …tractBasePtr AvgPool3DGradInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in AvgPool3DGradInfer()
|
| /third_party/typescript/tests/baselines/reference/ |
| D | nonPrimitiveInFunction.js | 8 var primitive: boolean; variable
28 var primitive; variable
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | norm_fusion.cc | 110 PrimitiveCPtr primitive = nullptr; in CreateNormNode() local
277 int CommonShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive… in CommonShapeSizeInfer()
282 …andDimsShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in ExpandDimsShapeSizeInfer()
287 …edSliceShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in StridedSliceShapeSizeInfer()
299 int MatMulShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive… in MatMulShapeSizeInfer()
304 int ReShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in ReShapeSizeInfer()
309 int StackSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in StackSizeInfer()
314 int SqueezeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in SqueezeSizeInfer()
323 int OneHotSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in OneHotSizeInfer()
328 int FillShapeSizeInfer(const std::vector<int> &in_shape_size, const schema::PrimitiveT &primitive) { in FillShapeSizeInfer()
[all …]
|