mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 14:24:22 +00:00
Move dtype size implementation to cpp file
This commit is contained in:
@@ -4,6 +4,19 @@
|
||||
|
||||
using namespace CUDANet;
|
||||
|
||||
size_t dtype_size(DType dtype) {
|
||||
switch (dtype)
|
||||
{
|
||||
case DType::FLOAT32:
|
||||
return 4;
|
||||
break;
|
||||
|
||||
default:
|
||||
throw std::runtime_error("Unknown DType");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
Tensor::Tensor(Shape shape, CUDANet::Backend* backend)
|
||||
: Tensor(shape, backend->get_default_dtype(), backend) {}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user