Change size_in_bytes argument type from int to size_t.

tensorflower-gardener · tensorflower-gardener · commit 64e41355fb8c · 2025-02-24T15:21:39.000-08:00
Other uses of it are size_t, so this makes it consistent.

PiperOrigin-RevId: 730609856
diff --git a/tensorflow/lite/experimental/litert/runtime/opencl/buffer.cc b/tensorflow/lite/experimental/litert/runtime/opencl/buffer.cc
@@ -28,7 +28,7 @@
 
 namespace litert {
 namespace cl {
-absl::Status CreateClBuffer(cl_context context, int size_in_bytes,
+absl::Status CreateClBuffer(cl_context context, size_t size_in_bytes,
                             bool read_only, void* data, cl_mem* result) {
   cl_mem_flags flags = read_only ? CL_MEM_READ_ONLY : CL_MEM_READ_WRITE;
   if (data) {
diff --git a/tensorflow/lite/experimental/litert/runtime/opencl/buffer.h b/tensorflow/lite/experimental/litert/runtime/opencl/buffer.h
@@ -72,7 +72,7 @@ class Buffer {
 
 Buffer CreateBufferShared(cl_mem buffer);
 
-absl::Status CreateClBuffer(cl_context context, int size_in_bytes,
+absl::Status CreateClBuffer(cl_context context, size_t size_in_bytes,
                             bool read_only, void* data, cl_mem* result);
 
 absl::Status CreateBuffer(size_t size_in_bytes, bool gpu_read_only,
