diff --git a/oneflow/core/operator/operator.cpp b/oneflow/core/operator/operator.cpp
index 0e14fd36529a5613ce81c1a7ed011b333f188371..7d92af7924b642e84ae05ba8a1b17b1cf1aa708d 100644
--- a/oneflow/core/operator/operator.cpp
+++ b/oneflow/core/operator/operator.cpp
@@ -121,7 +121,7 @@ Maybe<void> Operator::InferLogicalOutBlobDescs(
   auto* map = sbp_signature.mutable_bn_in_op2sbp_parallel();
   for (const auto& ibn : input_bns()) { (*map)[ibn].mutable_split_parallel()->set_axis(0); }
   for (const auto& obn : output_bns()) { (*map)[obn].mutable_split_parallel()->set_axis(0); }
-  return InferBlobDescsIf(BlobDesc4BnInOp, &parallel_ctx, &sbp_signature, [](OpContext*) {});
+  return InferOutBlobDescsIf(BlobDesc4BnInOp, &parallel_ctx, &sbp_signature, [](OpContext*) {});
 }
 
 Maybe<void> Operator::InferBlobDescsIf(