message: Fix regression caused by cl/191020868: Re-use materialized shapes for other broadcast gradient shape nodes.
diff: @@ -552,7 +552,6 @@ Status ConstantFolding::MaterializeBroadcastGradientArgs(
  const DataType type = node.attr().at("T").type();
  NodeDef* out[2];
  - bool created_const = false;
  for (int j = 0; j < 2; ++j) {
  int reduction_indices = reduce_dims[j].size();
  Tensor value(type, TensorShape({reduction_indices}));
  @@ -576,11 +575,9 @@ Status ConstantFolding::MaterializeBroadcastGradientArgs(
  AddControlDependency(node.name(), graph_, node_map_.get());
  *out[j]->add_input() = ctrl_dep;
  node_map_->AddOutput(NodeName(ctrl_dep), const_name);
  - created_const = true;
  }
  }
  - if (created_const) {
  const std::set<NodeDef*> outputs = node_map_->GetOutputs(node.name());
  for (NodeDef* output : outputs) {
  for (int k = 0; k < output->input_size(); ++k) {
  @@ -592,7 +589,6 @@ Status ConstantFolding::MaterializeBroadcastGradientArgs(
  }
  }
  }
  - }
  return Status::OK();
  }
  