message: Add ability to stop shape inference from propagating caller callee constants.
diff: @@ -1627,8 +1627,10 @@ LogicalResult InferModuleShape(ModuleOp module, int64_t max_iterations) {
  return success();
  }
  int64_t producer = producer_or.ValueOrDie();
  + // TODO(jpienaar): Clean up propagate_caller_callee_constants if it is no
  + // longer needed.
  ShapeInference context(producer, module.getContext(),
  - /*propagate_caller_callee_constants=*/true);
  + /*propagate_caller_callee_constants=*/false);
  if (auto main = module.lookupSymbol<mlir::FuncOp>("main"))
  context.enqueue(main);
  for (auto func : module.getOps<FuncOp>()) context.enqueue(func);
  