mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-23 14:54:28 +00:00
Refactor CUDA kernels and tensor operations for type generality
This commit is contained in:
@@ -80,6 +80,10 @@ Tensor::~Tensor() {
|
||||
}
|
||||
}
|
||||
|
||||
DType Tensor::get_dtype() const {
|
||||
return dtype;
|
||||
}
|
||||
|
||||
size_t Tensor::numel() const {
|
||||
return total_elms;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user