fix

Former-commit-id: bd6bb915b862fe49b266304531d3eae59e7eaa10

oneflow/core/operator/multiple_gather_op.cpp

@@ -37,7 +37,7 @@ void MultipleGatherOp::InferBlobDescs(
 
 bool MultipleGatherOp::IsInputBlobAllowedModelSplit(const std::string& ibn) const {
   CHECK(std::find(input_bns().begin(), input_bns().end(), ibn) != input_bns().end());
-  return false;
+  return ibn == "in";
 }
 
 void MultipleGatherOp::InferOutputBlobModelSplitAxis(