Searched defs:split_info (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 92 bool CalSplitOutputShape(int64_t splited_axis_value, const SplitInfo *split_info, in CalSplitOutputShape()
124 …Shape(const std::vector<std::vector<ShapeVector>> &node_in_out_shapes, const SplitInfo *split_info, in CalSplitInShape()
221 …const FuncGraphPtr &func_graph, const std::vector<AnfNodePtr> &conv_nodes, SplitInfo *split_info) { in UpdateSplitInfo()
333 … const std::vector<AnfNodePtr> &conv_outputs, const SplitInfo &split_info, in CreateOutputsOfConcat()
386 … std::vector<AnfNodePtr> *split_outputs, const SplitInfo &split_info, in CreateOutputsOfSplitWithOverlap()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info_register.cc | 29 std::ostringstream split_info; in ToString() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | flash_attention_score_info.cc | 974 std::vector<int64_t> split_info = GetSplitIdAndRank(); in ReplaceActualSeqLenForSplitSeqInTnd() local
1030 std::vector<int64_t> split_info = GetSplitIdAndRank(); in ReplaceNodeInputOrAttrs() local
1217 std::vector<int64_t> split_info = GetSplitIdAndRank(); in ComputeReplaceGraphForLoadBalance() local
|