diff options
Diffstat (limited to 'src/armnn/layers/ConvertFp32ToBf16Layer.hpp')
-rw-r--r-- | src/armnn/layers/ConvertFp32ToBf16Layer.hpp | 42 |
1 files changed, 0 insertions, 42 deletions
diff --git a/src/armnn/layers/ConvertFp32ToBf16Layer.hpp b/src/armnn/layers/ConvertFp32ToBf16Layer.hpp deleted file mode 100644 index 71de4fbcda..0000000000 --- a/src/armnn/layers/ConvertFp32ToBf16Layer.hpp +++ /dev/null @@ -1,42 +0,0 @@ -// -// Copyright © 2020 Arm Ltd and Contributors. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include <Layer.hpp> - -namespace armnn -{ - -/// This layer converts data type Float32 to BFloat16. -class ConvertFp32ToBf16Layer : public Layer -{ -public: - /// Makes a workload for the ConvertFp32ToBf16Layer type. - /// @param [in] factory The workload factory which will create the workload. - /// @return A pointer to the created workload, or nullptr if not created. - virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override; - - /// Creates a dynamically-allocated copy of this layer. - /// @param [in] graph The graph into which this layer is being cloned. - ConvertFp32ToBf16Layer* Clone(Graph& graph) const override; - - /// Check if the input tensor shape(s) - /// will lead to a valid configuration of @ref ConvertFp32ToBf16Layer. - /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validated. - void ValidateTensorShapesFromInputs() override; - - void ExecuteStrategy(IStrategy& strategy) const override; - -protected: - /// Constructor to create a ConvertFp32ToBf16Layer. - /// @param [in] name Optional name for the layer. - ConvertFp32ToBf16Layer(const char* name); - - /// Default destructor - ~ConvertFp32ToBf16Layer() = default; -}; - -} // namespace |