/*
// Copyright (c) 2017-2018 Intel Corporation
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
*/
#pragma once

#include "inference_engine.hpp"
#include "mkldnn/mkldnn_generic_primitive.hpp"
#include "mkldnn/mkldnn_extension.hpp"
#include <vector>

namespace IECustomExtension {

class TensorflowLayer : public InferenceEngine::MKLDNNPlugin::IMKLDNNGenericPrimitive {
public:
    explicit TensorflowLayer(const InferenceEngine::CNNLayerPtr& layer) : _layer(layer) {
        // try to check parameters availability in constructor
        auto genLayer = reinterpret_cast<InferenceEngine::GenericLayer*>(_layer.get());
        const std::string protobuf = genLayer->GetParamAsString("protobuf");
        const std::string input_nodes_names = genLayer->GetParamAsString("input_nodes_names");
        const std::string output_tensors_names = genLayer->GetParamAsString("output_tensors_names");
    }

    ~TensorflowLayer() override {}

    std::vector<InferenceEngine::MKLDNNPlugin::MKLDNNGenericFormats> GetSupportedFormats() noexcept override;

    void Execute() noexcept override;

private:
    InferenceEngine::CNNLayerPtr _layer;
};

}  // namespace IECustomExtension
