fp16 implementation, TODO deallocate in LayerWgs
This commit is contained in:
+3
-1
@@ -58,7 +58,9 @@ public:
|
||||
|
||||
dataDim_t input_dim;
|
||||
dataDim_t getOutputDim();
|
||||
|
||||
bool fp16;
|
||||
};
|
||||
|
||||
}
|
||||
#endif //NETWORK_H
|
||||
#endif //NETWORK_H
|
||||
|
||||
Reference in New Issue
Block a user