Home
last modified time | relevance | path

Searched refs:sub_ops (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h119 Ops sub_ops() const { return sub_ops_; } in sub_ops() function
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc626 if (!op_info->sub_ops().empty()) { in SplitTensor()
627 auto sub_ops = op_info->sub_ops(); in SplitTensor() local
628 for (size_t i = 0; i < sub_ops.size(); i++) { in SplitTensor()
629 if (!sub_ops.at(i).empty()) { in SplitTensor()
630 InsertGetTensorSliceOp(sub_ops.at(i).at(0), next_node, func_graph, index, SUB); in SplitTensor()