Home
last modified time | relevance | path

Searched refs:ConvertFp32ToFp16Layer (Results 1 – 18 of 18) sorted by relevance

/external/armnn/src/armnn/layers/
DConvertFp32ToFp16Layer.cpp16 ConvertFp32ToFp16Layer::ConvertFp32ToFp16Layer(const char* name) in ConvertFp32ToFp16Layer() function in armnn::ConvertFp32ToFp16Layer
21 std::unique_ptr<IWorkload> ConvertFp32ToFp16Layer::CreateWorkload(const IWorkloadFactory& factory) … in CreateWorkload()
29 ConvertFp32ToFp16Layer* ConvertFp32ToFp16Layer::Clone(Graph& graph) const in Clone()
31 return CloneBase<ConvertFp32ToFp16Layer>(graph, GetName()); in Clone()
34 void ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs() in ValidateTensorShapesFromInputs()
50 void ConvertFp32ToFp16Layer::ExecuteStrategy(IStrategy& strategy) const in ExecuteStrategy()
DConvertFp32ToFp16Layer.hpp13 class ConvertFp32ToFp16Layer : public Layer class
23 ConvertFp32ToFp16Layer* Clone(Graph& graph) const override;
35 ConvertFp32ToFp16Layer(const char* name);
38 ~ConvertFp32ToFp16Layer() = default;
/external/armnn/src/armnn/test/optimizations/
DOptimizeInverseConversionsTests.cpp27 graph.InsertNewLayer<armnn::ConvertFp32ToFp16Layer>(output->GetInputSlot(0), "convert1");
34 graph.InsertNewLayer<armnn::ConvertFp32ToFp16Layer>(output->GetInputSlot(0), "convert4");
37 &IsLayerOfType<armnn::ConvertFp32ToFp16Layer>,
40 … &IsLayerOfType<armnn::ConvertFp32ToFp16Layer>, &IsLayerOfType<armnn::OutputLayer>));
DFp32NetworkToFp16ConverterTests.cpp43 &IsLayerOfType<armnn::ConvertFp32ToFp16Layer>,
/external/armnn/src/armnn/
DNetworkUtils.cpp79 std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer) in InsertConvertFp32ToFp16LayersAfter()
83 std::vector<ConvertFp32ToFp16Layer*> convertLayers; in InsertConvertFp32ToFp16LayersAfter()
97 ConvertFp32ToFp16Layer* convertLayer = in InsertConvertFp32ToFp16LayersAfter()
98 graph.InsertNewLayer<ConvertFp32ToFp16Layer>(outputSlot, name.c_str()); in InsertConvertFp32ToFp16LayersAfter()
DNetworkUtils.hpp18 std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer);
DNetwork.cpp895 std::vector<ConvertFp32ToFp16Layer*> convertFp32ToFp16Layers; in AttemptBackendAssignment()
948 for (ConvertFp32ToFp16Layer* convertLayer : convertFp32ToFp16Layers) in AttemptBackendAssignment()
2127 return m_Graph->AddLayer<ConvertFp32ToFp16Layer>(name); in AddConvertFp32ToFp16Layer()
/external/armnn/src/armnn/optimizations/
DOptimizeInverseConversions.hpp42 …OptimizeForConnection<ConvertFp16ToFp32Layer, ConvertFp32ToFp16Layer, OptimizeInverseConversionsIm…
44 …OptimizeForConnection<ConvertFp32ToFp16Layer, ConvertFp16ToFp32Layer, OptimizeInverseConversionsIm…
/external/armnn/src/backends/reference/test/
DRefLayerSupportTests.cpp118 …ool result = IsConvertLayerSupportedTests<armnn::RefWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
128 …ool result = IsConvertLayerSupportedTests<armnn::RefWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
139 …ool result = IsConvertLayerSupportedTests<armnn::RefWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
/external/armnn/src/backends/cl/test/
DClLayerSupportTests.cpp96 …bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
106 …bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
117 …bool result = IsConvertLayerSupportedTests<armnn::ClWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
/external/armnn/src/backends/neon/test/
DNeonLayerSupportTests.cpp72 …ol result = IsConvertLayerSupportedTests<armnn::NeonWorkloadFactory, armnn::ConvertFp32ToFp16Layer,
/external/armnn/src/armnn/test/
DOptimizerTests.cpp416 &IsLayerOfType<armnn::ConvertFp32ToFp16Layer>,
419 &IsLayerOfType<armnn::ConvertFp32ToFp16Layer>,
DShapeInferenceTests.cpp280 CreateGraphAndRunTest<ConvertFp32ToFp16Layer>({{ 5, 7, 6, 2 }}, {{ 5, 7, 6, 2 }}, "floor");
/external/armnn/
DAndroid.bp138 "src/armnn/layers/ConvertFp32ToFp16Layer.cpp",
DCMakeLists.txt214 src/armnn/layers/ConvertFp32ToFp16Layer.hpp
215 src/armnn/layers/ConvertFp32ToFp16Layer.cpp
DLICENSE.spdx7466 FileName: ./src/armnn/layers/ConvertFp32ToFp16Layer.cpp
7473 FileName: ./src/armnn/layers/ConvertFp32ToFp16Layer.hpp
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp1985ConvertFp32ToFp16Layer* const layer = graph.AddLayer<ConvertFp32ToFp16Layer>("Fp32ToFp16Converter"… in CreateConvertFp32ToFp16WorkloadTest()
/external/armnn/docs/
D02_operator_list.dox701 <td rowspan="3">ConvertFp32ToFp16Layer