Searched refs:kPrimSplit (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | split_one_pass.cc | 42 if (!CheckPrimitiveType(node, prim::kPrimSplit)) { in Run()
|
| D | transpose_strategy.cc | 418 …if (CheckPrimitiveType(cnode, prim::kPrimSplit) || CheckPrimitiveType(cnode, prim::kPrimQuantDType… in CanFusionIfInsert()
470 {prim::kPrimConcat->name(), ChangeCommonOp}, {prim::kPrimSplit->name(), ChangeCommonOp}, in ChangeOpAxis()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 167 if (op_name == prim::kPrimSplit->name()) { in NeedSkip()
207 if (op_name != prim::kPrimSplit->name() && op_name != prim::kPrimSplitV->name()) { in Process()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 67 {prim::kPrimSplit->name(), {{0}, {kAllPositions}}},
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | glu_fusion.cc | 51 split_prim_ = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSplit>); in Init()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_inputs_adjust.cc | 183 …if (opt::CheckPrimitiveType(cnode, prim::kPrimSplit) && cnode->inputs().size() == split_inputs_siz… in Run()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 131 {prim::kPrimSplit, {InferImplSplit, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 65 auto split = VectorRef({GetPrim(prim::kPrimSplit), sigmoid}); in DefineGruCellPattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 116 … prim::kPrimSplit, prim::kPrimTranspose, in CanOpFullQuantized()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 235 inline const PrimitivePtr kPrimSplit = std::make_shared<Primitive>("Split"); variable
|