Searched refs:InsertGetTensorSliceOp (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 372 void InsertGetTensorSliceOp(const Operator &op, const CNodePtr &node, const FuncGraphPtr &func_grap… in InsertGetTensorSliceOp() function
625 InsertGetTensorSliceOp(op, next_node, func_graph, index, SPLIT_TENSOR); in SplitTensor()
630 InsertGetTensorSliceOp(sub_ops.at(i).at(0), next_node, func_graph, index, SUB); in SplitTensor()
2410 InsertGetTensorSliceOp(op, grad_sens_node, func_graph, 1, SPLIT_SENS); in SplitSens()
|