Support overriding memory allocation functions

- Let users provide their own memory management functions for XNNPACK

PiperOrigin-RevId: 281355722
diff --git a/test/convolution-operator-tester.h b/test/convolution-operator-tester.h
index cf97a08..f6a092d 100644
--- a/test/convolution-operator-tester.h
+++ b/test/convolution-operator-tester.h
@@ -643,7 +643,7 @@
         });
 
       // Create, setup, run, and destroy Convolution operator.
-      ASSERT_EQ(xnn_status_success, xnn_initialize());
+      ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
       xnn_operator_t convolution_op = nullptr;
 
       ASSERT_EQ(xnn_status_success,
@@ -800,7 +800,7 @@
       }
 
       // Create, setup, run, and destroy Convolution operator.
-      ASSERT_EQ(xnn_status_success, xnn_initialize());
+      ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
       xnn_operator_t convolution_op = nullptr;
 
       ASSERT_EQ(xnn_status_success,
@@ -966,7 +966,7 @@
       }
 
       // Create, setup, run, and destroy Convolution operator.
-      ASSERT_EQ(xnn_status_success, xnn_initialize());
+      ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
       xnn_operator_t convolution_op = nullptr;
 
       xnn_status status = xnn_create_convolution2d_nchw_f32(
@@ -1109,7 +1109,7 @@
         });
 
       // Create, setup, and run Convolution operator once.
-      ASSERT_EQ(xnn_status_success, xnn_initialize());
+      ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
       xnn_operator_t convolution_op = nullptr;
 
       ASSERT_EQ(xnn_status_success,
@@ -1325,7 +1325,7 @@
       }
 
       // Create, setup, and run Convolution operator once.
-      ASSERT_EQ(xnn_status_success, xnn_initialize());
+      ASSERT_EQ(xnn_status_success, xnn_initialize(nullptr /* allocator */));
       xnn_operator_t convolution_op = nullptr;
 
       ASSERT_EQ(xnn_status_success,