Searched defs:slice_cnode (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 194 …ATUS SlicePreposePass::SwapSliceWithPreceed(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, in SwapSliceWithPreceed()
243 ValueNodePtr SlicePreposePass::CopySliceValueNode(const CNodePtr &slice_cnode) { in CopySliceValueNode()
266 auto slice_cnode = graph->NewCNode(inputs); in InsertSlice() local
275 STATUS SlicePreposePass::VerifySliceAttrs(const CNodePtr &slice_cnode, const int dim) { in VerifySliceAttrs()
318 STATUS SlicePreposePass::SliceParamDeBroadcast(const CNodePtr &slice_cnode, const std::vector<int64… in SliceParamDeBroadcast()
512 int64_t SlicePreposePass::GetReshapeAbnormalIndexOut(const CNodePtr &slice_cnode, in GetReshapeAbnormalIndexOut()
561 … SlicePreposePass::PreposeWithNormalReshape(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, in PreposeWithNormalReshape()
626 …ePreposePass::CreateSlice1ForReshapePrepose(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, in CreateSlice1ForReshapePrepose()
666 …ePreposePass::CreateSlice2ForReshapePrepose(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, in CreateSlice2ForReshapePrepose()
710 …licePreposePass::PreposeWithAbnormalReshape(const FuncGraphPtr &graph, const CNodePtr &slice_cnode, in PreposeWithAbnormalReshape()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | slice_activation_in_cell_share_recompute.cc | 127 …CNodePtr slice_cnode = CreateStridedSliceCNode(slice_begin, slice_end, slice_strides, activation_c… in CreateSliceNode() local
231 auto slice_cnode = CreateSliceNode(slice_pos_node, groups); in SliceReuseRecomputedActivationNodes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | grad_accumulation_utils.cc | 49 auto slice_cnode = node->cast<CNodePtr>(); in TagMicroBatchStart() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | slice_activation_in_recompute.cc | 148 CNodePtr slice_cnode = CreateStridedSliceCNode(slice_begin, slice_end, slice_strides, node); in InsertSliceAllGatherNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 1556 auto slice_cnode = pair.first->cast<CNodePtr>(); in CanMergeConcatSlice() local
2980 auto slice_cnode = need_replace_op_lists[row][col]; in MergeOpBeforeInterleaveSlice() local
2997 auto slice_cnode = need_replace_op_lists[row][col]; in MergeOpBeforeInterleaveSlice() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | flash_attention_fusion.cc | 435 auto slice_cnode = func_graph->NewCNode(slice_prim_c, inputs); in CreateSliceCNode() local
|