Skip to content

Commit a226ec1

Browse files
authored
Update EWLayer.hpp
1 parent 6062a29 commit a226ec1

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

include/layers_oneDNN/EWLayer.hpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ class EwLayerOneDnn : public Layer {
3737
void initialize_onednn(const Shape& shape, Type data_type);
3838
dnnl::algorithm get_algorithm() const;
3939
void validate_input(const std::vector<Tensor>& input) const;
40+
dnnl::memory::data_type EwLayerOneDnn::get_dnnl_data_type(Type type) const;
4041

4142
std::string func_;
4243
float alpha_;
@@ -49,4 +50,4 @@ class EwLayerOneDnn : public Layer {
4950
bool initialized_ = false;
5051
};
5152

52-
} // namespace it_lab_ai
53+
} // namespace it_lab_ai

0 commit comments

Comments
 (0)