diff --git a/foxi/onnxifi_ext.h b/foxi/onnxifi_ext.h
index 9f03684..85cf846 100644
--- a/foxi/onnxifi_ext.h
+++ b/foxi/onnxifi_ext.h
@@ -135,6 +135,10 @@ typedef ONNXIFI_CHECK_RESULT onnxStatus
   (ONNXIFI_ABI* onnxReleaseTraceEventsFunction)(
     onnxTraceEventList* traceEvents);
 
+typedef ONNXIFI_CHECK_RESULT onnxStatus
+  (ONNXIFI_ABI* onnxGetCurrentBatchSizeFunction)(
+    int64_t* currentBatchSize);
+
 /**
  * A combination of onnxSetIO and onnxRunGraph, functionally equals to first run
  * onnxSetIO(graph, inputsCount, inputDescriptors, outputsCount,