diff options
Diffstat (limited to 'inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h')
-rw-r--r-- | inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h index bde51127a..e206799f4 100644 --- a/inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h +++ b/inference-engine/src/mkldnn_plugin/nodes/mkldnn_eltwise_node.h @@ -26,11 +26,14 @@ public: bool isSum(); bool isUnitScales(); + void initOptimalPrimitiveDescriptor() override; private: static Register<MKLDNNEltwiseNode> reg; InferenceEngine::EltwiseLayer::eOperation op; std::vector<float> sum_scales; + + template <typename T0, typename T1> void ref_eltwise(int in0, int in1); }; } // namespace MKLDNNPlugin |