Searched refs:kPrimGather (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | gather_v2_ds_fission.cc | 94 …ePtr> gatherv2_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimGather->name())), pad, in CreateGatherV2Ds()
153 VectorRef pattern({prim::kPrimGather, Xs}); in DefinePattern()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 42 Register(prim::kPrimGather->name(), {2}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | gather.cc | 98 REGISTER_PRIMITIVE_EVAL_IMPL(Gather, prim::kPrimGather, GatherInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | inputs_adjust.cc | 119 } else if (opt::CheckPrimitiveType(node, prim::kPrimGather)) { in Run()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 69 {prim::kPrimGather, {3}}, in InConvertWhiteList()
90 {prim::kPrimGather, {3}}, {prim::kPrimReshape, {2}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | selection_ops_declare.cc | 79 REG_ADPT_DESC(GatherV2D, prim::kPrimGather->name(), ADPT_DESC(GatherV2D))
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 50 const auto kGather = prim::kPrimGather->name(); in GetDependsFormMap()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | weight_quantizer.cc | 440 } else if (opt::CheckPrimitiveType(cnode, prim::kPrimGather)) { in DoQuantize()
|
| D | quantize_util.cc | 113 … prim::kPrimGather, prim::kPrimLayerNormFusion, in CanOpFullQuantized()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 481 if (IsPrimitiveCNode(ele.first, prim::kPrimGather)) { in FindGatherV2FromSparseGatherV2()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 336 auto prim = prim::kPrimGather; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 180 inline const PrimitivePtr kPrimGather = std::make_shared<Primitive>("Gather"); variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1625 if (node->IsApply(prim::kPrimGather)) { in ExportCNode()
|