updated typos and misaligned types

This commit is contained in:
Alejandro Saucedo 2020-08-18 21:24:23 +01:00
parent 014f15d552
commit f8859c7fd6
4 changed files with 5 additions and 24 deletions

View file

@ -27,7 +27,7 @@ class Tensor
Tensor();
Tensor(std::vector<uint32_t> shape, TensorTypes tensorType = TensorTypes::eDevice);
Tensor(std::array<uint32_t, KP_MAX_DIM_SIZE> shape, TensorTypes tensorType = TensorTypes::eDevice);
~Tensor();
@ -39,7 +39,7 @@ class Tensor
// Getter functions
std::vector<uint32_t> data();
uint32_t size();
std::array<uint32_t, KP_MAX_DIM_SIZE> shape();
std::vector<uint32_t> shape();
TensorTypes tensorType();
bool isInit();