Add Tensor<>::size()

7b135645 · Po-Yen, Chen · 4643f0e6 · 7b135645
Commit 7b135645 authored Sep 14, 2022 by Po-Yen, Chen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

library/include/ck/library/utility/host_tensor.hpp library/include/ck/library/utility/host_tensor.hpp +2 -0

No files found.
--- a/library/include/ck/library/utility/host_tensor.hpp
+++ b/library/include/ck/library/utility/host_tensor.hpp
@@ -442,6 +442,8 @@ struct Tensor

    typename Data::const_pointer data() const { return mData.data(); }

+    typename Data::size_type size() const { return mData.size(); }
+
    Descriptor mDesc;
    Data mData;
 };