Home
last modified time | relevance | path

Searched refs:primitive (Results 1 – 25 of 2017) sorted by relevance

12345678910>>...81

/third_party/mindspore/mindspore/lite/mindir/include/
Dmindir.h43 ActivationType MindIR_Activation_GetActivationType(ConstPrimitivePtr primitive);
44 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type);
45 float MindIR_Activation_GetAlpha(ConstPrimitivePtr primitive);
46 void MindIR_Activation_SetAlpha(PrimitivePtr *primitive, float alpha);
47 float MindIR_Activation_GetMinVal(ConstPrimitivePtr primitive);
48 void MindIR_Activation_SetMinVal(PrimitivePtr *primitive, float min_val);
49 float MindIR_Activation_GetMaxVal(ConstPrimitivePtr primitive);
50 void MindIR_Activation_SetMaxVal(PrimitivePtr *primitive, float max_val);
51 bool MindIR_Activation_GetApproximate(ConstPrimitivePtr primitive);
52 void MindIR_Activation_SetApproximate(PrimitivePtr *primitive, bool approximate);
[all …]
/third_party/mindspore/mindspore/core/abstract/
Dinfer_functions.h44 AbstractBasePtr InferImplPooling(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
46 AbstractBasePtr InferImplPoolingGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
48 AbstractBasePtr InferImplFusedSparseAdam(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
50 AbstractBasePtr InferImplBatchNorm(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
52 AbstractBasePtr InferImplConv2D(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
54 AbstractBasePtr InferImplBiasAdd(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
56 AbstractBasePtr InferImplBiasAddGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
58 AbstractBasePtr InferImplHSigmoid(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
60 AbstractBasePtr InferImplHSigmoidGrad(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
62 AbstractBasePtr InferImplBpropCut(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
[all …]
Dprim_others.cc36 AbstractBasePtr InferImplIdentity(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplIdentity() argument
39 CheckArgsSize(primitive->name(), args_spec_list, 1); in InferImplIdentity()
43 AbstractBasePtr InferImplEnvGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvGetItem() argument
45 MS_EXCEPTION_IF_NULL(primitive); in InferImplEnvGetItem()
47 CheckArgsSize(primitive->name(), args_spec_list, 3); in InferImplEnvGetItem()
76 AbstractBasePtr InferImplEnvSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvSetItem() argument
79 CheckArgsSize(primitive->name(), args_spec_list, 3); in InferImplEnvSetItem()
94 AbstractBasePtr InferImplEnvAdd(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplEnvAdd() argument
97 CheckArgsSize(primitive->name(), args_spec_list, 2); in InferImplEnvAdd()
157 AbstractBasePtr InferImplStateSetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplStateSetItem() argument
[all …]
Dprim_structures.cc35 AbstractBasePtr InferImplMakeDict(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeDict() argument
38 const std::string op_name = primitive->name(); in InferImplMakeDict()
65 AbstractBasePtr InferImplMakeKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeKwarg() argument
68 const std::string op_name = primitive->name(); in InferImplMakeKwarg()
81 AbstractBasePtr InferImplExtractKwarg(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplExtractKwarg() argument
84 const std::string op_name = primitive->name(); in InferImplExtractKwarg()
103 AbstractBasePtr InferImplMakeSlice(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplMakeSlice() argument
106 CheckArgsSize(primitive->name(), args_spec_list, 3); in InferImplMakeSlice()
217 AbstractBasePtr InferImplTupleGetItem(const AnalysisEnginePtr &, const PrimitivePtr &primitive, in InferImplTupleGetItem() argument
219 return InferTupleOrListGetItem<AbstractTuple>(primitive->name(), args_spec_list); in InferImplTupleGetItem()
[all …]
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir.cc37 ActivationType MindIR_Activation_GetActivationType(ConstPrimitivePtr primitive) { in MindIR_Activation_GetActivationType() argument
38 if (primitive != nullptr) { in MindIR_Activation_GetActivationType()
39 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetActivationType()
53 void MindIR_Activation_SetActivationType(PrimitivePtr *primitive, ActivationType activation_type) { in MindIR_Activation_SetActivationType() argument
54 if (primitive != nullptr && *primitive != nullptr) { in MindIR_Activation_SetActivationType()
55 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetActivationType()
67 *primitive = ret_value; in MindIR_Activation_SetActivationType()
71 float MindIR_Activation_GetAlpha(ConstPrimitivePtr primitive) { in MindIR_Activation_GetAlpha() argument
72 if (primitive != nullptr) { in MindIR_Activation_GetAlpha()
73 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetAlpha()
[all …]
Dmindir_nnrt_lite_graph_to_model.cc177 std::vector<int8_t> ConvertActivation(PrimitivePtr primitive) { in ConvertActivation() argument
178 if (primitive != nullptr) { in ConvertActivation()
179 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertActivation()
201 std::vector<int8_t> ConvertAddFusion(PrimitivePtr primitive) { in ConvertAddFusion() argument
202 if (primitive != nullptr) { in ConvertAddFusion()
203 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertAddFusion()
220 std::vector<int8_t> ConvertArgMaxFusion(PrimitivePtr primitive) { in ConvertArgMaxFusion() argument
221 if (primitive != nullptr) { in ConvertArgMaxFusion()
222 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertArgMaxFusion()
242 std::vector<int8_t> ConvertAvgPoolFusion(PrimitivePtr primitive) { in ConvertAvgPoolFusion() argument
[all …]
/third_party/mindspore/mindspore/lite/src/common/
Dprim_util.cc28 int GetPrimitiveType(const void *primitive, int schema_version) { in GetPrimitiveType() argument
29 if (primitive == nullptr) { in GetPrimitiveType()
34 return static_cast<const schema::v0::Primitive *>(primitive)->value_type(); in GetPrimitiveType()
37 return static_cast<const schema::Primitive *>(primitive)->value_type(); in GetPrimitiveType()
40 const char *GetPrimitiveTypeName(const void *primitive, int schema_version) { in GetPrimitiveTypeName() argument
41 if (primitive == nullptr) { in GetPrimitiveTypeName()
46 …return schema::v0::EnumNamePrimitiveType(static_cast<const schema::v0::Primitive *>(primitive)->va… in GetPrimitiveTypeName()
49 …return schema::EnumNamePrimitiveType(static_cast<const schema::Primitive *>(primitive)->value_type… in GetPrimitiveTypeName()
58 bool IsPartialNode(const void *primitive, int schema_version) { in IsPartialNode() argument
59 MS_CHECK_TRUE_MSG(primitive != nullptr, false, "primtive cannot be nullptr"); in IsPartialNode()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/operator/
Dops_front_infer_function.h24 AbstractBasePtr InferImplHasType(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
26 …actBasePtr InferImplBroadcastGradientArgs(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
28 AbstractBasePtr InferImplListMap(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
30 AbstractBasePtr InferImplListReduce(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
32 AbstractBasePtr InferImplTupleReversed(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
34 AbstractBasePtr InferImplReduceShape(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
36 AbstractBasePtr InferImplTupleDiv(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
38 AbstractBasePtr InferImplTuple2Array(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
40 AbstractBasePtr InferImplShapeMul(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
42 AbstractBasePtr InferImplTupleEqual(const AnalysisEnginePtr &, const PrimitivePtr &primitive,
[all …]
/third_party/mindspore/mindspore/lite/test/ut/tools/converter/parser/tflite/
Dtflite_arithmetic_parser_test.cc31 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
32 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_AddFusion) << "w… in TEST_F()
44 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
45 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_SubFusion) << "w… in TEST_F()
57 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
58 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_MulFusion) << "w… in TEST_F()
70 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
71 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_DivFusion) << "w… in TEST_F()
82 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
83 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_FloorDiv) << "wr… in TEST_F()
[all …]
Dtflite_activation_parser_test.cc30 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
31 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "… in TEST_F()
35 ASSERT_NE(meta_graph->nodes.front()->primitive->value.AsActivation(), nullptr); in TEST_F()
36 auto val = meta_graph->nodes.front()->primitive->value.AsActivation(); in TEST_F()
48 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
49 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "… in TEST_F()
53 ASSERT_NE(meta_graph->nodes.front()->primitive->value.AsActivation(), nullptr); in TEST_F()
54 auto val = meta_graph->nodes.front()->primitive->value.AsActivation(); in TEST_F()
66 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
67 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_Activation) << "… in TEST_F()
[all …]
Dtflite_reduce_parser_test.cc30 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
31 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_ReduceFusion) <<… in TEST_F()
35 ASSERT_NE(meta_graph->nodes.front()->primitive->value.AsReduceFusion(), nullptr); in TEST_F()
36 auto val = meta_graph->nodes.front()->primitive->value.AsReduceFusion(); in TEST_F()
50 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
51 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_ReduceFusion) <<… in TEST_F()
55 ASSERT_NE(meta_graph->nodes.front()->primitive->value.AsReduceFusion(), nullptr); in TEST_F()
56 auto val = meta_graph->nodes.front()->primitive->value.AsReduceFusion(); in TEST_F()
70 ASSERT_NE(meta_graph->nodes.front()->primitive.get(), nullptr); in TEST_F()
71 …ASSERT_EQ(meta_graph->nodes.front()->primitive->value.type, schema::PrimitiveType_ReduceFusion) <<… in TEST_F()
[all …]
/third_party/protobuf/src/google/protobuf/util/internal/
Dprotostream_objectsource_test.cc193 Primitive primitive; in PrepareRepeatedPrimitive() local
194 primitive.add_rep_fix32(3201); in PrepareRepeatedPrimitive()
195 primitive.add_rep_fix32(0); in PrepareRepeatedPrimitive()
196 primitive.add_rep_fix32(3202); in PrepareRepeatedPrimitive()
197 primitive.add_rep_u32(3203); in PrepareRepeatedPrimitive()
198 primitive.add_rep_u32(0); in PrepareRepeatedPrimitive()
199 primitive.add_rep_i32(0); in PrepareRepeatedPrimitive()
200 primitive.add_rep_i32(3204); in PrepareRepeatedPrimitive()
201 primitive.add_rep_i32(3205); in PrepareRepeatedPrimitive()
202 primitive.add_rep_sf32(3206); in PrepareRepeatedPrimitive()
[all …]
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc49 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSmoothL1LossParameter() local
50 auto value = primitive->value_as_SmoothL1Loss(); in PopulateSmoothL1LossParameter()
52 p->op_parameter_.type_ = primitive->value_type(); in PopulateSmoothL1LossParameter()
64 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateSmoothL1LossGradParameter() local
65 auto value = primitive->value_as_SmoothL1LossGrad(); in PopulateSmoothL1LossGradParameter()
67 p->op_parameter_.type_ = primitive->value_type(); in PopulateSmoothL1LossGradParameter()
79 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateApplyMomentumParameter() local
80 auto value = primitive->value_as_ApplyMomentum(); in PopulateApplyMomentumParameter()
81 p->op_parameter_.type_ = primitive->value_type(); in PopulateApplyMomentumParameter()
93 auto primitive = static_cast<const schema::Primitive *>(prim); in PopulateBCEParameter() local
[all …]
Dtrain_populate_parameter_v0.cc37 OpParameter *DefaultPopulateParameter(const void *primitive) { in DefaultPopulateParameter() argument
38 if (primitive == nullptr) { in DefaultPopulateParameter()
42 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in DefaultPopulateParameter()
94 OpParameter *PopulateSmoothL1LossParameter(const void *primitive) { in PopulateSmoothL1LossParameter() argument
95 if (primitive == nullptr) { in PopulateSmoothL1LossParameter()
99 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSmoothL1LossParameter()
114 OpParameter *PopulateSmoothL1LossGradParameter(const void *primitive) { in PopulateSmoothL1LossGradParameter() argument
115 if (primitive == nullptr) { in PopulateSmoothL1LossGradParameter()
119 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSmoothL1LossGradParameter()
134 OpParameter *PopulateApplyMomentumParameter(const void *primitive) { in PopulateApplyMomentumParameter() argument
[all …]
/third_party/mindspore/mindspore/core/ops/
Dneighborexchange.cc44 void CheckAttr(const PrimitivePtr &primitive, const std::string &shape_attr_name, in CheckAttr() argument
46 MS_EXCEPTION_IF_NULL(primitive); in CheckAttr()
50 auto attr = primitive->GetAttr(shape_attr_name); in CheckAttr()
66 auto attr = primitive->GetAttr(rank_ids_attr_name); in CheckAttr()
75 …MS_EXCEPTION(ValueError) << "Invalid " << primitive->name() << " attr " << shape_attr_name << " si… in CheckAttr()
81 void Check(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in Check() argument
82 MS_EXCEPTION_IF_NULL(primitive); in Check()
83 auto prim_name = primitive->name(); in Check()
84 CheckAttr(primitive, kRecvShapes, kRecvRankIds); in Check()
85 CheckAttr(primitive, kSendShapes, kSendRankIds); in Check()
[all …]
Ddiag.cc27 abstract::ShapePtr DiagInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr>… in DiagInferShape() argument
28 MS_EXCEPTION_IF_NULL(primitive); in DiagInferShape()
31 primitive->name()); in DiagInferShape()
37 TypePtr PartInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args… in PartInferType() argument
38 MS_EXCEPTION_IF_NULL(primitive); in PartInferType()
40 …ndConvertUtils::CheckTensorTypeValid("input type", x_dtype, common_valid_types, primitive->name()); in PartInferType()
43 AbstractBasePtr DiagInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in DiagInfer() argument
45 MS_EXCEPTION_IF_NULL(primitive); in DiagInfer()
46 auto prim_name = primitive->name(); in DiagInfer()
51 …return abstract::MakeAbstract(DiagInferShape(primitive, input_args), PartInferType(primitive, inpu… in DiagInfer()
Ddiag_part.cc29 abstract::ShapePtr DiagPartInferShape(const PrimitivePtr &primitive, const std::vector<AbstractBase… in DiagPartInferShape() argument
30 MS_EXCEPTION_IF_NULL(primitive); in DiagPartInferShape()
31 auto op_name = primitive->name(); in DiagPartInferShape()
47 TypePtr DiagPartInferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_… in DiagPartInferType() argument
48 MS_EXCEPTION_IF_NULL(primitive); in DiagPartInferType()
50 …ndConvertUtils::CheckTensorTypeValid("input type", x_dtype, common_valid_types, primitive->name()); in DiagPartInferType()
53 AbstractBasePtr DiagPartInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in DiagPartInfer() argument
55 MS_EXCEPTION_IF_NULL(primitive); in DiagPartInfer()
56 auto prim_name = primitive->name(); in DiagPartInfer()
61 …return abstract::MakeAbstract(DiagPartInferShape(primitive, input_args), DiagPartInferType(primiti… in DiagPartInfer()
Dhshrink.cc29 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() argument
30 MS_EXCEPTION_IF_NULL(primitive); in InferShape()
31 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, 1L, primitive->name()); in InferShape()
35 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
36 MS_EXCEPTION_IF_NULL(primitive); in InferType()
37 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, 1, primitive->name()); in InferType()
40 primitive->name()); in InferType()
44 AbstractBasePtr HShrinkInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in HShrinkInfer() argument
46 return std::make_shared<abstract::AbstractTensor>(InferType(primitive, input_args), in HShrinkInfer()
47 InferShape(primitive, input_args)->shape()); in HShrinkInfer()
Dones_like.cc30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() argument
31 auto op_name = primitive->name(); in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
37 MS_EXCEPTION_IF_NULL(primitive); in InferType()
38 auto op_name = primitive->name(); in InferType()
47 AbstractBasePtr OnesLikeInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in OnesLikeInfer() argument
49 MS_EXCEPTION_IF_NULL(primitive); in OnesLikeInfer()
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in OnesLikeInfer()
Dzeros_like.cc30 abstract::ShapePtr InferShape(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &in… in InferShape() argument
31 auto op_name = primitive->name(); in InferShape()
36 TypePtr InferType(const PrimitivePtr &primitive, const std::vector<AbstractBasePtr> &input_args) { in InferType() argument
37 auto op_name = primitive->name(); in InferType()
45 AbstractBasePtr ZerosLikeInfer(const abstract::AnalysisEnginePtr &, const PrimitivePtr &primitive, in ZerosLikeInfer() argument
47 MS_EXCEPTION_IF_NULL(primitive); in ZerosLikeInfer()
49 CheckAndConvertUtils::CheckInputArgs(input_args, kEqual, input_num, primitive->name()); in ZerosLikeInfer()
50 …return abstract::MakeAbstract(InferShape(primitive, input_args), InferType(primitive, input_args)); in ZerosLikeInfer()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dweight_quantizer.cc62 const PrimitivePtr &primitive) { in SetAbstract() argument
65 MS_CHECK_TRUE_MSG(primitive != nullptr, RET_NULL_PTR, "primitive is nullptr."); in SetAbstract()
81 auto quant_param_holder = GetCNodeQuantHolder(primitive); in SetAbstract()
89 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoConvQuantize() local
90 if (primitive == nullptr) { in DoConvQuantize()
116 …status = MixedBitQuantFilter(tensor_info, primitive, QuantType_QUANT_WEIGHT, WeightQuantType::MIXE… in DoConvQuantize()
119 …status = FixedBitQuantFilter<int8_t>(tensor_info, primitive, QuantType_QUANT_WEIGHT, quant_max_, q… in DoConvQuantize()
122 …status = FixedBitQuantFilter<int16_t>(tensor_info, primitive, QuantType_QUANT_WEIGHT, quant_max_, … in DoConvQuantize()
131 status = SetAbstract(tensor_info, param_node, primitive); in DoConvQuantize()
149 auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0)); in DoMulQuantize() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_util.cc149 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomCount() local
150 MS_EXCEPTION_IF_NULL(primitive); in GetHcomCount()
151 if (primitive->GetAttr(kAttrRankSize) != nullptr) { in GetHcomCount()
152 rank_size = GetValue<int64_t>(primitive->GetAttr(kAttrRankSize)); in GetHcomCount()
190 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomOperationType() local
191 MS_EXCEPTION_IF_NULL(primitive); in GetHcomOperationType()
192 if (primitive->GetAttr(kAttrOp) == nullptr) { in GetHcomOperationType()
196 auto hcom_op_type = GetValue<std::string>(primitive->GetAttr(kAttrOp)); in GetHcomOperationType()
215 auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); in GetHcomRootId() local
216 MS_EXCEPTION_IF_NULL(primitive); in GetHcomRootId()
[all …]
/third_party/mindspore/mindspore/lite/test/ut/src/
Dscheduler_test.cc44 split->primitive = std::make_unique<mindspore::schema::PrimitiveT>(); in TEST_F()
45 split->primitive->value.type = mindspore::schema::PrimitiveType_Split; in TEST_F()
46 auto primitive = new mindspore::schema::SplitT; in TEST_F() local
47 primitive->output_num = 2; in TEST_F()
48 primitive->axis = 3; in TEST_F()
49 split->primitive->value.value = primitive; in TEST_F()
55 abs1->primitive = std::make_unique<mindspore::schema::PrimitiveT>(); in TEST_F()
56 abs1->primitive->value.type = mindspore::schema::PrimitiveType_Abs; in TEST_F()
58 abs1->primitive->value.value = abs1_primitive; in TEST_F()
64 cons1->primitive = std::make_unique<mindspore::schema::PrimitiveT>(); in TEST_F()
[all …]
/third_party/node/deps/npm/node_modules/es-to-primitive/
DREADME.md1 # es-to-primitive <sup>[![Version Badge][npm-version-svg]][package-url]</sup>
18 var toPrimitive = require('es-to-primitive');
37 [package-url]: https://npmjs.org/package/es-to-primitive
38 [npm-version-svg]: http://versionbadg.es/ljharb/es-to-primitive.svg
39 [travis-svg]: https://travis-ci.org/ljharb/es-to-primitive.svg
40 [travis-url]: https://travis-ci.org/ljharb/es-to-primitive
41 [deps-svg]: https://david-dm.org/ljharb/es-to-primitive.svg
42 [deps-url]: https://david-dm.org/ljharb/es-to-primitive
43 [dev-deps-svg]: https://david-dm.org/ljharb/es-to-primitive/dev-status.svg
44 [dev-deps-url]: https://david-dm.org/ljharb/es-to-primitive#info=devDependencies
[all …]
/third_party/mesa3d/src/gallium/auxiliary/util/
Du_prim.h184 u_vertices_per_prim(enum pipe_prim_type primitive) in u_vertices_per_prim() argument
186 switch(primitive) { in u_vertices_per_prim()
230 u_decomposed_prims_for_vertices(enum pipe_prim_type primitive, int vertices) in u_decomposed_prims_for_vertices() argument
232 switch (primitive) { in u_decomposed_prims_for_vertices()
277 u_reduced_prims_for_vertices(enum pipe_prim_type primitive, int vertices) in u_reduced_prims_for_vertices() argument
279 switch (primitive) { in u_reduced_prims_for_vertices()
282 return u_decomposed_prims_for_vertices(primitive, vertices) * 2; in u_reduced_prims_for_vertices()
284 primitive = PIPE_PRIM_TRIANGLE_FAN; in u_reduced_prims_for_vertices()
287 return u_decomposed_prims_for_vertices(primitive, vertices); in u_reduced_prims_for_vertices()
341 u_stream_outputs_for_vertices(enum pipe_prim_type primitive, unsigned nr) in u_stream_outputs_for_vertices() argument
[all …]

12345678910>>...81