updated typos and misaligned types
This commit is contained in:
parent
014f15d552
commit
f8859c7fd6
4 changed files with 5 additions and 24 deletions
|
|
@ -7,15 +7,11 @@ Tensor::Tensor() {
|
|||
this->mTensorType = TensorTypes::eDevice;
|
||||
}
|
||||
|
||||
Tensor::Tensor(std::vector<uint32_t> shape, TensorTypes tensorType)
|
||||
Tensor::Tensor(std::array<uint32_t, KP_MAX_DIM_SIZE> shape, TensorTypes tensorType)
|
||||
{
|
||||
SPDLOG_DEBUG("Kompute Tensor init with data");
|
||||
|
||||
if (shape.size() > KP_MAX_DIM_SIZE) {
|
||||
spdlog::warn("Kompute Tensor created with more dimensions than supported. Max: {}, Provided: {}.", KP_MAX_DIM_SIZE, shape.size());
|
||||
}
|
||||
|
||||
std::copy_n(shape.begin(), KP_MAX_DIM_SIZE, this->mShape.begin());
|
||||
this->mShape = shape;
|
||||
this->mTensorType = tensorType;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue