Searched defs:dyn_shape_tmp_fix (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.cc | 279 bool dyn_shape_tmp_fix) { in PrepareFlashAttentionScore()
360 …areFillV2(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareFillV2()
406 bool dyn_shape_tmp_fix) { in PreparePropagateBatchMatMul()
496 …reBiasAdd(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareBiasAdd()
521 …aParallel(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareDataParallel()
564 …idedSlice(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareStridedSlice()
641 …reSoftMax(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareSoftMax()
672 …LayerNorm(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareLayerNorm()
718 …reRmsNorm(const std::shared_ptr<OperatorInfo> &op, Dimensions basic_stra, bool dyn_shape_tmp_fix) { in PrepareRmsNorm()
743 …epareOneHot(const std::shared_ptr<OperatorInfo> &op, Dimensions strategy, bool dyn_shape_tmp_fix) { in PrepareOneHot()
[all …]
|
| D | rec_graph.h | 104 bool dyn_shape_tmp_fix = false; variable
|
| D | rec_parse_graph.cc | 451 const bool dyn_shape_tmp_fix) { in EliminateGraph()
|
| D | rec_cost.cc | 158 bool hasBeenSplitted(const Graph::NodeType &node, const bool dyn_shape_tmp_fix) { in hasBeenSplitted()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_auto_parallel.cc | 1387 …Graph(const std::vector<AnfNodePtr> &all_nodes, const FuncGraphPtr &root, bool dyn_shape_tmp_fix) { in ReInitCostGraph()
1451 bool dyn_shape_tmp_fix = false; in ParallelStrategyRecSearch() local
|