message: Add XlaReduceWindow as a supported op for TPU compilation
  This is a short-term fix for the op but longer term, we may want to generalize this somehow based on Ken's input.
diff: @@ -112,6 +112,8 @@ void AddSupportedControlFlowOps(MLIRContext* context,
  OperationName(TF::IfRegionOp::getOperationName(), context));
  supported_ops->insert(
  OperationName(TF::WhileRegionOp::getOperationName(), context));
  + supported_ops->insert(
  + OperationName(TF::XlaReduceWindowOp::getOperationName(), context));
  supported_ops->insert(
  OperationName(TF::YieldOp::getOperationName(), context));
  }
  