Searched refs:varianceVector (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/armnn/optimizations/ |
| D | FuseBatchNorm.hpp | 84 std::vector<T> varianceVector(varBuffer, varBuffer + varTensor.GetNumElements()); in Run() local
93 … T mult = gammaVector[cOut] / static_cast<T>(sqrtf(varianceVector[cOut] + epsilon)); in Run()
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 | 151 std::vector<T> varianceVector = GetVector<T>(outputDimensionSizes[3], 1.0f, 0.1f); in CreateNetwork() local
156 … ConstTensor variance(TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), varianceVector); in CreateNetwork()
|
| D | FuseActivationTests.cpp | 218 std::vector<T> varianceVector = GetVector<T>(GetOutputShape()[3], 1.0f, 0.1f); in AddReceiverLayer() local
224 … ConstTensor variance(TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), varianceVector); in AddReceiverLayer()
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 877 std::vector<float> varianceVector = { 1 }; variable
881 …ensor variance(TensorInfo(1, outputChannelSize, DataType::Float32, 0.0f, 0, true), varianceVector);
|