Fix all reduce mem sharing (#1986)

* fix all reduce mem sharing

* ByteSizeOfDataContentField=>ByteSizeOfBlobBody

oneflow/core/graph/reduce_comp_task_node_if.cpp

@@ -7,7 +7,7 @@
 namespace oneflow {
 
 int64_t InferRegstSize(const RegstDesc& regst) {
-  return RtBlobDesc(*(regst.GetBlobDesc(GenPackedLbi()))).ByteSizeOfDataContentField();
+  return RtBlobDesc(*(regst.GetBlobDesc(GenPackedLbi()))).ByteSizeOfBlobBody();
 }
 
 TaskNode* ReduceCompTaskNodeIf::FindPredReduceTaskNodeIf(std::function<bool(TaskNode*)> predicate) {

oneflow/core/graph/task_graph.cpp

@@ -372,7 +372,7 @@ void TaskGraph::AddReduceNoBwForwardNodeOverlapingCtrlEdges() {
 void TaskGraph::EnableMemSharingInReduceStruct() {
   auto GetSuccReduceTaskNode = [](TaskNode* pred) {
     std::vector<TaskNode*> nodes;
-    pred->ForEachNodeOnOutEdge([&](TaskNode* succ) {
+    pred->ForEachNodeOnOutDataEdge([&](TaskNode* succ) {
       if (dynamic_cast<ReduceCompTaskNodeIf*>(succ) != nullptr) { nodes.push_back(succ); }
     });
     return nodes;