1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
8 #include <mkldnn_node.h>
13 namespace MKLDNNPlugin {
15 class MKLDNNDepthwiseNode : public MKLDNNNode {
17 MKLDNNDepthwiseNode(InferenceEngine::CNNLayerPtr layer, const mkldnn::engine& eng);
18 ~MKLDNNDepthwiseNode() override = default;
20 void createDescriptor(const std::vector<InferenceEngine::TensorDesc>& inputDesc,
21 const std::vector<InferenceEngine::TensorDesc>& outputDesc) override;
22 void initOptimalPrimitiveDescriptor() override;
23 void getSupportedDescriptors() override;
24 void createPrimitive() override;
25 bool created() const override;
27 mkldnn::algorithm getAlgorithm() {
47 bool initialized = false;
49 static Register<MKLDNNDepthwiseNode> reg;
51 mkldnn::algorithm algorithm;
52 size_t realWeightSize = 0;
53 size_t realBiasSize = 0;
58 } // namespace MKLDNNPlugin