Searched refs:betaVector (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/armnn/optimizations/ |
| D | FuseBatchNorm.hpp | 81 std::vector<T> betaVector (betaBuffer, betaBuffer + betaTensor.GetNumElements()); in Run() local
149 … sqrtf(varianceVector[cOut] + epsilon)) + betaVector[cOut]; in Run()
160 … sqrtf(varianceVector[cOut] + epsilon)) + betaVector[cOut]; in Run()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | FuseBatchNormTests.cpp | 148 std::vector<T> betaVector = GetVector<T>(outputDimensionSizes[3], 0.0f, 0.2f); in CreateNetwork() local
153 ConstTensor beta (TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), betaVector); in CreateNetwork()
|
| D | FuseActivationTests.cpp | 215 std::vector<T> betaVector = GetVector<T>(GetOutputShape()[3], 0.0f, 0.2f); in AddReceiverLayer() local
221 ConstTensor beta(TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), betaVector); in AddReceiverLayer()
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 874 std::vector<float> betaVector = { 0.1f }; variable
878 … ConstTensor beta(TensorInfo(1, outputChannelSize, DataType::Float32, 0.0f, 0, true), betaVector);
|