fix

Former-commit-id: fe7f367d4410e66d9559566ccc175bf0632ecf0f

oneflow/core/operator/boxing_align_broadcast_reshape_op.cpp

@@ -52,7 +52,8 @@ void BoxingAlignBroadcastReshapeOp::InferOutputBlobModelSplitAxis(
     std::function<int32_t*(const std::string&)> ModelSplitAxis4BnInOp,
     std::function<int32_t(const std::string&)> ShapeNumAxes4BnInOp,
     const ParallelContext* parallel_context) const {
-  UNIMPLEMENTED();
+  CHECK_EQ(parallel_context->policy(), kDataParallel);
+  NaiveInferOutputBlobModelSplitAxis(ModelSplitAxis4BnInOp, ShapeNumAxes4BnInOp, parallel_context);
 }
 
 REGISTER_OP(OperatorConf::kBoxingAlignBroadcastReshapeConf, BoxingAlignBroadcastReshapeOp);

oneflow/core/operator/multiple_gather_op.cpp

@@ -44,7 +44,8 @@ void MultipleGatherOp::InferOutputBlobModelSplitAxis(
     std::function<int32_t*(const std::string&)> ModelSplitAxis4BnInOp,
     std::function<int32_t(const std::string&)> ShapeNumAxes4BnInOp,
     const ParallelContext* parallel_context) const {
-  UNIMPLEMENTED();
+  CHECK_EQ(parallel_context->policy(), kDataParallel);
+  NaiveInferOutputBlobModelSplitAxis(ModelSplitAxis4BnInOp, ShapeNumAxes4BnInOp, parallel_context);
 }
 
 REGISTER_OP(OperatorConf::kMultipleGatherConf, MultipleGatherOp);

oneflow/core/operator/resize_reshape_op.cpp

@@ -30,7 +30,8 @@ void ResizeReshapeOp::InferOutputBlobModelSplitAxis(
     std::function<int32_t*(const std::string&)> ModelSplitAxis4BnInOp,
     std::function<int32_t(const std::string&)> ShapeNumAxes4BnInOp,
     const ParallelContext* parallel_context) const {
-  UNIMPLEMENTED();
+  CHECK_EQ(parallel_context->policy(), kDataParallel);
+  NaiveInferOutputBlobModelSplitAxis(ModelSplitAxis4BnInOp, ShapeNumAxes4BnInOp, parallel_context);
 }
 
 REGISTER_OP(OperatorConf::kResizeReshapeConf, ResizeReshapeOp);