Searched refs:gammaVector (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/armnn/optimizations/ |
| D | FuseBatchNorm.hpp | 82 std::vector<T> gammaVector (gammaBuffer, gammaBuffer + gammaTensor.GetNumElements()); in Run() local
93 … T mult = gammaVector[cOut] / static_cast<T>(sqrtf(varianceVector[cOut] + epsilon)); in Run()
148 … fusedBiasVector[cOut] = ((gammaVector[cOut] * (biasVector[cOut] - meanVector[cOut])) / in Run()
159 … fusedBiasVector[cOut] = ((gammaVector[cOut] * (biasVector[cOut] - meanVector[cOut])) / in Run()
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | FuseBatchNormTests.cpp | 149 std::vector<T> gammaVector = GetVector<T>(outputDimensionSizes[3], 0.5f, 0.1f); in CreateNetwork() local
154 ConstTensor gamma (TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), gammaVector); in CreateNetwork()
|
| D | FuseActivationTests.cpp | 216 std::vector<T> gammaVector = GetVector<T>(GetOutputShape()[3], 0.5f, 0.1f); in AddReceiverLayer() local
222 ConstTensor gamma(TensorInfo(1, outputChannelSize, ArmnnType, 0.0f, 0, true), gammaVector); in AddReceiverLayer()
|
| /external/armnn/src/armnn/test/ |
| D | OptimizerTests.cpp | 875 std::vector<float> gammaVector = { 0.5f }; variable
879 …ConstTensor gamma(TensorInfo(1, outputChannelSize, DataType::Float32, 0.0f, 0, true), gammaVector);
|