blob: aa7f892915a01595568da0367d0c45685e169ab4 [file] [log] [blame]
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// See LICENSE file in the project root for full license information.
//
#pragma once
#include <Layer.hpp>
namespace armnn
{
class FloorLayer : public Layer
{
public:
virtual std::unique_ptr<IWorkload> CreateWorkload(const Graph& graph,
const IWorkloadFactory& factory) const override;
FloorLayer* Clone(Graph& graph) const override;
void ValidateTensorShapesFromInputs() override;
protected:
FloorLayer(const char* name);
~FloorLayer() = default;
};
} // namespace