Home
last modified time | relevance | path

Searched defs:slice_inputs (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dunsorted_segment_sum_d_fission.cc100 std::vector<AnfNodePtr> slice_inputs = {NewValueNode(std::make_shared<Primitive>(kSliceOpName)), in CreateSlice() local
Dunsorted_segment_sum_fission.cc85 …std::vector<AnfNodePtr> slice_inputs = {NewValueNode(std::make_shared<Primitive>(kSliceOpName)), n… in CreateSlice() local
Dgather_v2_ds_fission.cc145 …std::vector<AnfNodePtr> slice_inputs = {NewValueNode(std::make_shared<Primitive>(kSliceOpName)), g… in CreateSlice() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc1273 std::vector<AnfNodePtr> slice_inputs = {new_slice_vnode, arithmetic_cnode->input(i), in PreposeWithArithmetic() local
1313 …std::vector<AnfNodePtr> slice_inputs = {new_slice_vnode, arithmetic_cnode->input(i), begin_paramet… in PreposeWithArithmetic() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dneighbor_exchange_v2_fusion.cc201 …std::vector<AnfNodePtr> slice_inputs = {NewValueNode(std::make_shared<Primitive>(kSliceOpName)), s… in CreateSplitNode() local