Searched refs:gammaTensor (Results 1 – 3 of 3) sorted by relevance
78 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in BatchNormTestImpl() local84 descriptor.m_Gamma = &gammaTensor; in BatchNormTestImpl()92 AllocateAndCopyDataToITensorHandle(&gammaTensor, gamma.data()); in BatchNormTestImpl()167 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in BatchNormTestNhwcImpl() local172 AllocateAndCopyDataToITensorHandle(&gammaTensor, &gamma[0]); in BatchNormTestNhwcImpl()179 data.m_Gamma = &gammaTensor; in BatchNormTestNhwcImpl()657 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in CompareBatchNormTest() local662 AllocateAndCopyDataToITensorHandle(&gammaTensor, &gamma[0]); in CompareBatchNormTest()669 data.m_Gamma = &gammaTensor; in CompareBatchNormTest()
84 ScopedTensorHandle gammaTensor(tensorInfo);89 AllocateAndCopyDataToITensorHandle(&gammaTensor, gamma.data());96 data.m_Gamma = &gammaTensor;
49 …ConstTensor gammaTensor(batchNormLayer->m_Gamma->GetTensorInfo(), batchNormLayer->m_Gamma->Map(tru… in Run() local76 const auto* gammaBuffer = static_cast<const T*>(gammaTensor.GetMemoryArea()); in Run()82 std::vector<T> gammaVector (gammaBuffer, gammaBuffer + gammaTensor.GetNumElements()); in Run()