We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6062a29 commit a226ec1Copy full SHA for a226ec1
1 file changed
include/layers_oneDNN/EWLayer.hpp
@@ -37,6 +37,7 @@ class EwLayerOneDnn : public Layer {
37
void initialize_onednn(const Shape& shape, Type data_type);
38
dnnl::algorithm get_algorithm() const;
39
void validate_input(const std::vector<Tensor>& input) const;
40
+ dnnl::memory::data_type EwLayerOneDnn::get_dnnl_data_type(Type type) const;
41
42
std::string func_;
43
float alpha_;
@@ -49,4 +50,4 @@ class EwLayerOneDnn : public Layer {
49
50
bool initialized_ = false;
51
};
52
-} // namespace it_lab_ai
53
+} // namespace it_lab_ai
0 commit comments