Searched refs:GET_NEXT (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 73 if (prim->name() == GET_NEXT || prim->name() == VIRTUAL_OUTPUT) { in IsParallelCareNode()
|
| D | step_auto_parallel.cc | 167 …STRIDEDSLICE, GET_NEXT, CAST, NEG, SQUARE, BATCH_MATMUL, EXPAND_DIMS, SQUEEZE, SPARSE_GATHERV2, TI… in IsSplittableOperator()
|
| D | step_parallel.cc | 61 static const std::set<std::string> INVALID_LOSS_OPS = {GET_NEXT, VIRTUALLOSS, LOAD, UPDATESTATE};
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | get_next_info.cc | 217 replace_op_ = {std::make_pair(GET_NEXT, args)}; in InferReplaceOps()
|
| D | ops_utils.h | 336 constexpr char GET_NEXT[] = "GetNext"; variable
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.cc | 616 if (iter_graph != SIZE_MAX && ops[iter_ops]->type() != GET_NEXT) { in GeneratePartitionedOperatorStrategy()
|