Home
last modified time | relevance | path

Searched defs:prim (Results 1 – 25 of 924) sorted by relevance

12345678910>>...37

/third_party/mindspore/tests/ut/cpp/operator/
Dops_test.cc28 namespace prim { namespace
38 auto prim = std::make_shared<Primitive>(prim::kScalarAdd); in TEST_F() local
43 auto prim = std::make_shared<Primitive>(prim::kScalarSub); in TEST_F() local
48 auto prim = std::make_shared<Primitive>(prim::kScalarMul); in TEST_F() local
53 auto prim = std::make_shared<Primitive>(prim::kScalarDiv); in TEST_F() local
58 auto prim = std::make_shared<Primitive>(prim::kScalarMod); in TEST_F() local
63 auto prim = std::make_shared<Primitive>(prim::kScalarPow); in TEST_F() local
68 auto prim = std::make_shared<Primitive>(prim::kScalarTrunc); in TEST_F() local
73 auto prim = std::make_shared<Primitive>(prim::kScalarFloor); in TEST_F() local
78 auto prim = std::make_shared<Primitive>(prim::kScalarUadd); in TEST_F() local
[all …]
Dprim2func_test.cc28 namespace prim { namespace
37 auto prim = std::make_shared<Primitive>(prim::kScalarAdd); in TEST_F() local
52 auto prim = std::make_shared<Primitive>("scalar_exp"); in TEST_F() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_arithmetic_operation_parser.cc52 auto prim = std::make_unique<ops::AddFusion>(); in Parse() local
58 auto prim = std::make_unique<ops::SubFusion>(); in Parse() local
64 auto prim = std::make_unique<ops::DivFusion>(); in Parse() local
70 auto prim = std::make_unique<ops::MulFusion>(); in Parse() local
76 auto prim = std::make_unique<ops::Equal>(); in Parse() local
82 auto prim = std::make_unique<ops::Less>(); in Parse() local
88 auto prim = std::make_unique<ops::Greater>(); in Parse() local
94 auto prim = std::make_unique<ops::Floor>(); in Parse() local
100 auto prim = std::make_unique<ops::Abs>(); in Parse() local
106 auto prim = std::make_unique<ops::ExpFusion>(); in Parse() local
[all …]
Donnx_activation_parser.cc29 auto prim = std::make_unique<ops::Activation>(); in Parse() local
37 auto prim = std::make_unique<ops::Activation>(); in Parse() local
52 auto prim = std::make_unique<ops::PReLUFusion>(); in Parse() local
109 auto prim = std::make_unique<ops::Activation>(); in Parse() local
123 auto prim = std::make_unique<ops::Activation>(); in Parse() local
131 auto prim = std::make_unique<ops::Activation>(); in Parse() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/
Dtflite_arithmetic_parser.cc58 auto prim = std::make_unique<ops::AddFusion>(); in Parse() local
74 auto prim = std::make_unique<ops::MulFusion>(); in Parse() local
92 auto prim = std::make_unique<ops::DivFusion>(); in Parse() local
110 auto prim = std::make_unique<ops::SubFusion>(); in Parse() local
128 auto prim = std::make_unique<ops::FloorDiv>(); in Parse() local
139 auto prim = std::make_unique<ops::FloorMod>(); in Parse() local
150 auto prim = std::make_unique<ops::PowFusion>(); in Parse() local
164 auto prim = std::make_unique<ops::SquaredDifference>(); in Parse() local
175 auto prim = std::make_unique<ops::Maximum>(); in Parse() local
186 auto prim = std::make_unique<ops::Minimum>(); in Parse() local
[all …]
Dtflite_custom_parser.cc77 auto prim = std::make_unique<ops::DetectionPostProcess>(); in DetectPostProcess() local
111 auto prim = std::make_unique<ops::AudioSpectrogram>(); in AudioSpectrogram() local
123 auto prim = std::make_unique<ops::Mfcc>(); in Mfcc() local
136 auto prim = std::make_unique<ops::CustomPredict>(); in Predict() local
145 auto prim = std::make_unique<ops::CustomNormalize>(); in Normalize() local
151 auto prim = std::make_unique<ops::CustomExtractFeatures>(); in ExtractFeatures() local
164 auto prim = std::make_unique<ops::Rfft>(); in Rfft() local
178 auto prim = std::make_unique<ops::FftReal>(); in FftReal() local
184 auto prim = std::make_unique<ops::FftImag>(); in FftImag() local
190 auto prim = std::make_unique<ops::Identity>(); in Identity() local
Dtflite_activation_parser.cc33 auto prim = std::make_unique<ops::Activation>(); in Parse() local
46 auto prim = std::make_unique<ops::Activation>(); in Parse() local
59 auto prim = std::make_unique<ops::Activation>(); in Parse() local
76 auto prim = std::make_unique<ops::PReLUFusion>(); in Parse() local
89 auto prim = std::make_unique<ops::Activation>(); in Parse() local
102 auto prim = std::make_unique<ops::Activation>(); in Parse() local
115 auto prim = std::make_unique<ops::Activation>(); in Parse() local
Dtflite_logical_parser.cc33 auto prim = std::make_unique<ops::LogicalAnd>(); in Parse() local
44 auto prim = std::make_unique<ops::LogicalNot>(); in Parse() local
55 auto prim = std::make_unique<ops::LogicalOr>(); in Parse() local
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir_nnrt_lite_graph_to_model.cc101 auto prim = static_cast<schema::Primitive *>(node->primitive_); in MindIR_LiteGraph_To_Model() local
179 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertActivation() local
203 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertAddFusion() local
222 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertArgMaxFusion() local
244 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertAvgPoolFusion() local
291 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertBatchToSpaceND() local
330 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertBiasAdd() local
348 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertCast() local
366 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertConcat() local
385 auto prim = static_cast<schema::Primitive *>(primitive); in ConvertConv2DFusion() local
[all …]
Dmindir.cc39 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetActivationType() local
55 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetActivationType() local
73 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetAlpha() local
87 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetAlpha() local
104 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetMinVal() local
118 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetMinVal() local
135 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetMaxVal() local
149 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetMaxVal() local
166 auto prim = static_cast<const schema::Primitive *>(primitive); in MindIR_Activation_GetApproximate() local
180 auto prim = static_cast<schema::Primitive *>(*primitive); in MindIR_Activation_SetApproximate() local
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Ditem_tuple_or_list_eliminate.h45 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsVNode})(node); in operator() local
46 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsVNode})(node); in operator() local
47 AnfVisitor::Match(prim::kPrimTupleSetItem, {IsCNode, IsVNode, IsNode})(node); in operator() local
48 AnfVisitor::Match(prim::kPrimListSetItem, {IsCNode, IsVNode, IsNode})(node); in operator() local
103 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsVNode})(node); in operator() local
104 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsVNode})(node); in operator() local
151 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsVNode, IsVNode})(node); in operator() local
152 AnfVisitor::Match(prim::kPrimListGetItem, {IsVNode, IsVNode})(node); in operator() local
203 AnfVisitor::Match(prim::kPrimTupleSetItem, {IsCNode, IsVNode, IsNode})(node); in operator() local
204 AnfVisitor::Match(prim::kPrimListSetItem, {IsCNode, IsVNode, IsNode})(node); in operator() local
[all …]
Dmerge_addn.h42 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(node); in operator() local
75 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(first_input); in Visit() local
95 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(last_input); in Visit() local
152 AnfVisitor::Match(prim::kPrimAddN, {IsCNode})(node); in operator() local
Dminmax_grad.h39 auto prim = GetValueNode<PrimitivePtr>(cnode->input(0)); in IsOriginMaxMinGrad() local
57 …AnfVisitor::Match(prim::kPrimTupleGetItem, {internal::IsOriginMaxMinGrad, IsValueNode<Int64Imm>})(… in operator() local
71 auto prim = GetValueNode<PrimitivePtr>(inputs[0]); in operator() local
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc42 OpParameter *PopulateSmoothL1LossParameter(const void *prim) { in PopulateSmoothL1LossParameter()
57 OpParameter *PopulateSmoothL1LossGradParameter(const void *prim) { in PopulateSmoothL1LossGradParameter()
72 OpParameter *PopulateApplyMomentumParameter(const void *prim) { in PopulateApplyMomentumParameter()
87 OpParameter *PopulateBCEParameter(const void *prim) { in PopulateBCEParameter()
100 OpParameter *PopulateBCEGradParameter(const void *prim) { in PopulateBCEGradParameter()
113 OpParameter *PopulateAdamParameter(const void *prim) { in PopulateAdamParameter()
128 OpParameter *PopulateSgdParameter(const void *prim) { in PopulateSgdParameter()
146 OpParameter *PopulateSparseSoftmaxCrossEntropyWithLogitsParameter(const void *prim) { in PopulateSparseSoftmaxCrossEntropyWithLogitsParameter()
162 OpParameter *PopulateSoftmaxCrossEntropyParameter(const void *prim) { in PopulateSoftmaxCrossEntropyParameter()
176 OpParameter *PopulateMaxPoolGradParameter(const void *prim) { in PopulateMaxPoolGradParameter()
[all …]
Dtrain_populate_parameter_v0.cc42 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in DefaultPopulateParameter() local
99 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSmoothL1LossParameter() local
119 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSmoothL1LossGradParameter() local
139 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateApplyMomentumParameter() local
157 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateBCEParameter() local
170 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateBCEGradParameter() local
183 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateAdamParameter() local
203 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSgdParameter() local
230 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulateSparseSoftmaxCrossEntropyParameter() local
268 auto *prim = static_cast<const schema::v0::Primitive *>(primitive); in PopulatePoolingGradParameter() local
[all …]
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dtf_arithmetic_parser.cc55 auto prim = std::make_unique<ops::AddFusion>(); in Parse() local
69 auto prim = std::make_unique<ops::SubFusion>(); in Parse() local
83 auto prim = std::make_unique<ops::MulFusion>(); in Parse() local
97 auto prim = std::make_unique<ops::DivFusion>(); in Parse() local
111 auto prim = std::make_unique<ops::Maximum>(); in Parse() local
125 auto prim = std::make_unique<ops::Minimum>(); in Parse() local
139 auto prim = std::make_unique<ops::Greater>(); in Parse() local
153 auto prim = std::make_unique<ops::GreaterEqual>(); in Parse() local
167 auto prim = std::make_unique<ops::Less>(); in Parse() local
181 auto prim = std::make_unique<ops::LessEqual>(); in Parse() local
[all …]
Dtf_logical_parser.cc31 auto prim = std::make_unique<ops::LogicalAnd>(); in Parse() local
44 auto prim = std::make_unique<ops::LogicalOr>(); in Parse() local
57 auto prim = std::make_unique<ops::LogicalNot>(); in Parse() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_activation_parser.cc25 auto prim = std::make_unique<ops::Activation>(); in Parse() local
41 auto prim = std::make_unique<ops::Activation>(); in Parse() local
49 auto prim = std::make_unique<ops::Activation>(); in Parse() local
57 auto prim = std::make_unique<ops::Activation>(); in Parse() local
65 auto prim = std::make_unique<ops::Activation>(); in Parse() local
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc71 bool MakeDfGraph(PrimitivePtr prim, unsigned int nparam) { in MakeDfGraph()
105 auto prim = std::make_shared<Primitive>("MaxPool"); in TEST_F() local
116 auto prim = prim::kPrimRelu; in TEST_F() local
174 auto prim = prim::kPrimConv2DBackpropInput; in TEST_F() local
202 auto prim = prim::kPrimConv2DBackpropFilter; in TEST_F() local
230 auto prim = prim::kPrimReluGrad; in TEST_F() local
253 auto prim = std::make_shared<Primitive>("BiasAdd"); in TEST_F() local
275 auto prim = prim::kPrimBiasAddGrad; in TEST_F() local
297 auto prim = std::make_shared<Primitive>("MaxPoolGradWithArgmax"); in TEST_F() local
325 auto prim = prim::kPrimConcat; in TEST_F() local
[all …]
/third_party/mesa3d/src/gallium/auxiliary/util/
Du_prim.h51 u_decomposed_prim(enum pipe_prim_type prim) in u_decomposed_prim()
76 u_reduced_prim(enum pipe_prim_type prim) in u_reduced_prim()
96 u_assembled_prim(enum pipe_prim_type prim) in u_assembled_prim()
118 u_prim_vertex_count(enum pipe_prim_type prim) in u_prim_vertex_count()
145 u_prims_for_vertices(enum pipe_prim_type prim, unsigned num) in u_prims_for_vertices()
/third_party/mindspore/mindspore/ccsrc/pybind_api/ir/
Dprimitive_py.cc351 auto prim = std::make_shared<PrimitivePy>(obj_adapter, prim_adapter); in Clone() local
409 auto prim = attached_primitive_.lock(); in AddPyAttr() local
431 auto prim = attached_primitive_.lock(); in DelPyAttr() local
438 auto prim = attached_primitive_.lock(); in GetAttrDict() local
452 auto prim = attached_primitive_.lock(); in set_prim_type() local
459 auto prim = attached_primitive_.lock(); in set_const_prim() local
466 auto prim = attached_primitive_.lock(); in set_const_input_indexes() local
474 auto prim = attached_primitive_.lock(); in set_signatures() local
482 auto prim = attached_primitive_.lock(); in set_hook() local
490 auto prim = attached_primitive_.lock(); in set_instance_name() local
[all …]
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_pipe_vbuf.c142 vbuf_tri(struct draw_stage *stage, struct prim_header *prim) in vbuf_tri()
156 vbuf_line(struct draw_stage *stage, struct prim_header *prim) in vbuf_line()
170 vbuf_point(struct draw_stage *stage, struct prim_header *prim) in vbuf_point()
186 vbuf_start_prim(struct vbuf_stage *vbuf, uint prim) in vbuf_start_prim()
268 vbuf_first_tri(struct draw_stage *stage, struct prim_header *prim) in vbuf_first_tri()
280 vbuf_first_line(struct draw_stage *stage, struct prim_header *prim) in vbuf_first_line()
292 vbuf_first_point(struct draw_stage *stage, struct prim_header *prim) in vbuf_first_point()
/third_party/mesa3d/src/mesa/tnl/
Dt_split_inplace.c47 const struct _mesa_prim *prim; member
115 struct _mesa_prim *prim = &split->dstprim[split->dstprim_nr++]; in next_outprim() local
124 const struct _mesa_prim *prim) in update_index_bounds()
137 const struct _mesa_prim *prim) in get_max_vertices()
159 const struct _mesa_prim *prim = &split->prim[i]; in split_prims() local
267 const struct _mesa_prim *prim, in _tnl_split_inplace()
/third_party/mesa3d/src/gallium/drivers/i915/
Di915_prim_emit.c118 emit_prim(struct draw_stage *stage, struct prim_header *prim, unsigned hwprim, in emit_prim()
158 setup_tri(struct draw_stage *stage, struct prim_header *prim) in setup_tri()
164 setup_line(struct draw_stage *stage, struct prim_header *prim) in setup_line()
170 setup_point(struct draw_stage *stage, struct prim_header *prim) in setup_point()
/third_party/mesa3d/src/mesa/state_tracker/
Dst_cb_feedback.c122 feedback_tri( struct draw_stage *stage, struct prim_header *prim ) in feedback_tri()
135 feedback_line( struct draw_stage *stage, struct prim_header *prim ) in feedback_line()
152 feedback_point( struct draw_stage *stage, struct prim_header *prim ) in feedback_point()
210 select_tri( struct draw_stage *stage, struct prim_header *prim ) in select_tri()
219 select_line( struct draw_stage *stage, struct prim_header *prim ) in select_line()
228 select_point( struct draw_stage *stage, struct prim_header *prim ) in select_point()

12345678910>>...37