Fixed tensor docstring according to suggestion

As for the title. Suggestion provided by axsaucedo Co-authored-by: Alejandro Saucedo <[email protected]> Signed-off-by: Andrea Calabrese <[email protected]>
KomputeProject · Aug 20, 2024 · 5f6777b · 5f6777b
1 parent 8112ed1
commit 5f6777b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/include/kompute/Tensor.hpp b/src/include/kompute/Tensor.hpp
@@ -100,7 +100,8 @@ class Tensor
      * Function to reserve memory on the tensor. This does not copy any data, it
      * just reserves memory, similarly to std::vector reserve() method.
      *
-     * @param newSize the new size for reservation
+     * @param elementTotalCount Total number of elements for new size
+     * @param elementMemorySize Memory size of element 
      */
     void reserve(uint32_t elementTotalCount, uint32_t elementMemorySize);