aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/optimizations/ConvertConstants.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/optimizations/ConvertConstants.hpp')
-rw-r--r--src/armnn/optimizations/ConvertConstants.hpp54
1 files changed, 0 insertions, 54 deletions
diff --git a/src/armnn/optimizations/ConvertConstants.hpp b/src/armnn/optimizations/ConvertConstants.hpp
index 54c14e5c89..7b2f1fd291 100644
--- a/src/armnn/optimizations/ConvertConstants.hpp
+++ b/src/armnn/optimizations/ConvertConstants.hpp
@@ -11,7 +11,6 @@
#include <armnn/backends/TensorHandle.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
-#include <BFloat16.hpp>
#include <Half.hpp>
namespace armnn
@@ -19,27 +18,6 @@ namespace armnn
namespace optimizations
{
-struct BFloat16ToFloat32
-{
- static void Func(std::shared_ptr<ConstTensorHandle>& handle)
- {
- const TensorInfo& info = handle->GetTensorInfo();
-
- if (info.GetDataType() == DataType::BFloat16)
- {
- std::vector<float> newValues(info.GetNumElements());
-
- armnnUtils::FloatingPointConverter::ConvertBFloat16ToFloat32(handle->GetConstTensor<BFloat16>(),
- info.GetNumElements(),
- newValues.data());
-
- TensorInfo newInfo(info.GetShape(), DataType::Float32, 0.0f, 0, true);
- ConstTensor newInput(newInfo, newValues);
- handle.reset(new ScopedTensorHandle(newInput));
- }
- }
-};
-
struct Float16ToFloat32
{
static void Func(std::shared_ptr<ConstTensorHandle>& handle)
@@ -61,27 +39,6 @@ struct Float16ToFloat32
}
};
-struct Float32ToBFloat16
-{
- static void Func(std::shared_ptr<ConstTensorHandle>& handle)
- {
- const TensorInfo& info = handle->GetTensorInfo();
-
- if (info.GetDataType() == DataType::Float32)
- {
- std::vector<BFloat16> newValues(info.GetNumElements());
-
- armnnUtils::FloatingPointConverter::ConvertFloat32ToBFloat16(handle->GetConstTensor<float>(),
- info.GetNumElements(),
- newValues.data());
-
- TensorInfo newInfo(info.GetShape(), DataType::BFloat16, 0.0f, 0, true);
- ConstTensor newInput(newInfo, newValues);
- handle.reset(new ScopedTensorHandle(newInput));
- }
- }
-};
-
struct Float32ToFloat16
{
static void Func(std::shared_ptr<ConstTensorHandle>& handle)
@@ -138,17 +95,6 @@ struct IsFloat16Layer
}
};
-struct IsBFloat16Layer
-{
- static bool Test(const Layer& layer)
- {
- return layer.GetDataType() == DataType::BFloat16;
- }
-};
-
-using ConvertConstantsBFloatToFloat = ConvertConstants<BFloat16ToFloat32, IsFloat32Layer>;
-using ConvertConstantsFloatToBFloat = ConvertConstants<Float32ToBFloat16, IsBFloat16Layer>;
-
using ConvertConstantsHalfToFloat = ConvertConstants<Float16ToFloat32, IsFloat32Layer>;
using ConvertConstantsFloatToHalf = ConvertConstants<Float32ToFloat16, IsFloat16Layer>;