Function SbpGetState

oneflow/api/python/symbol/sbp_symbol.cpp:62–72

}

Maybe<std::pair<std::string, int>> SbpGetState(const Symbol<SbpParallel>& sbp) {
  if (sbp->has_broadcast_parallel()) {
    return std::make_shared<std::pair<std::string, int>>("B", -1);
  } else if (sbp->has_partial_sum_parallel()) {
    return std::make_shared<std::pair<std::string, int>>("P", -1);
  } else if (sbp->has_split_parallel()) {
    return std::make_shared<std::pair<std::string, int>>("S", sbp->split_parallel().axis());
  } else {
    return Error::RuntimeError() << "Invalid sbp signature: " << sbp->DebugString();
  }
}

Maybe<Symbol<SbpParallel>> GetSbpFromState(const std::pair<std::string, int>& state) {
  if (state.first == "B") {