Method OptLocalParallel4BnInOp

oneflow/core/operator/operator.cpp:1087–1094

}

Maybe<const OptLocalParallel*> Operator::OptLocalParallel4BnInOp(
    const std::string& bn_in_op) const {
  CHECK_OR_RETURN(local_signature_) << "local signature not infered";
  const auto& map = local_signature_->bn_in_op2opt_local_parallel();
  const auto& iter = map.find(bn_in_op);
  CHECK_OR_RETURN(iter != map.end()) << "blob_name " << bn_in_op << " not found in local signature";
  return &iter->second;
}

OptLocalParallel* Operator::MutOptLocalParallel(const std::string& bn_in_op) {
  if (!local_signature_) { local_signature_.reset(new LocalSignature()); }