// | |
// Copyright © 2017 Arm Ltd. All rights reserved. | |
// SPDX-License-Identifier: MIT | |
// | |
#pragma once | |
#include <Layer.hpp> | |
namespace armnn | |
{ | |
class InputLayer : public BindableLayer | |
{ | |
public: | |
virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph, | |
const IWorkloadFactory& factory) const override; | |
InputLayer* Clone(Graph& graph) const override; | |
void ValidateTensorShapesFromInputs() override; | |
protected: | |
InputLayer(LayerBindingId id, const char* name); | |
~InputLayer() = default; | |
}; | |
} // namespace |