mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-12-22 22:34:22 +00:00
Refactor size calculations in layers and backend
This commit is contained in:
@@ -28,6 +28,7 @@ class Backend {
|
||||
std::optional<DType> default_dtype;
|
||||
public:
|
||||
|
||||
// Dtypes
|
||||
virtual bool supports_dtype(DType dtype) const = 0;
|
||||
virtual void set_default_dtype(DType dtype) = 0;
|
||||
virtual DType get_default_dtype() const = 0;
|
||||
|
||||
Reference in New Issue
Block a user