From 9ac4f15cb612fa560c3893803f700e5bc8295fe7 Mon Sep 17 00:00:00 2001 From: Andrea Calabrese <18246320+ThePseudo@users.noreply.github.com> Date: Sun, 18 Aug 2024 11:27:05 +0200 Subject: [PATCH] Fixed tensor docstring according to suggestion As for the title. Suggestion provided by axsaucedo Co-authored-by: Alejandro Saucedo --- src/include/kompute/Tensor.hpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/include/kompute/Tensor.hpp b/src/include/kompute/Tensor.hpp index 8c31aab2..3fd00244 100644 --- 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);