• Home
  • Raw
  • Download

Lines Matching full:beta

63     float beta,  in SimpleSoftmaxBaseTestImpl()  argument
94 data.m_Parameters.m_Beta = beta; in SimpleSoftmaxBaseTestImpl()
124 float beta) in SimpleSoftmaxTestImpl() argument
129 float x0[4] = { exp((0.f - 1.0f) * beta), exp((1.0f - 1.0f) * beta), in SimpleSoftmaxTestImpl()
130 exp((0.0f - 1.0f) * beta), exp((0.0f - 1.0f) * beta) }; in SimpleSoftmaxTestImpl()
132 float x1[4] = { exp((0.5f - 0.5f) * beta), exp((0.0f - 0.5f) * beta), in SimpleSoftmaxTestImpl()
133 exp((0.0f - 0.5f) * beta), exp((0.0f - 0.5f) * beta) }; in SimpleSoftmaxTestImpl()
145 … SimpleSoftmaxBaseTestImpl<ArmnnType, 2>(workloadFactory, memoryManager, tensorHandleFactory, beta, in SimpleSoftmaxTestImpl()
154 float beta, in SimpleSoftmaxTestImpl() argument
201 … SimpleSoftmaxBaseTestImpl<ArmnnType, 2>(workloadFactory, memoryManager, tensorHandleFactory, beta, in SimpleSoftmaxTestImpl()
210 float beta, in Simple3dSoftmaxTestImpl() argument
216 … SimpleSoftmaxBaseTestImpl<ArmnnType, 3>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxTestImpl()
225 float beta, in Simple4dSoftmaxTestImpl() argument
232 … SimpleSoftmaxBaseTestImpl<ArmnnType, 4>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxTestImpl()
243 float beta) in CompareSoftmaxTestImpl() argument
270 data.m_Parameters.m_Beta = beta; in CompareSoftmaxTestImpl()
319 float beta) in SimpleSoftmaxTest() argument
321 …ftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxTest()
328 float beta, in SimpleAxisSoftmaxTest() argument
332 tensorHandleFactory, beta, axis); in SimpleAxisSoftmaxTest()
339 float beta) in Simple3dSoftmaxTest() argument
342 …oftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxTest()
350 float beta, in Simple3dAxisSoftmaxTest() argument
437 …oftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dAxisSoftmaxTest()
445 float beta) in Simple4dSoftmaxTest() argument
449beta, data.inputShape, data.outputData, data.inputData); in Simple4dSoftmaxTest()
456 float beta, in Simple4dAxisSoftmaxTest() argument
611 beta, in Simple4dAxisSoftmaxTest()
622 float beta) in SimpleSoftmaxUint8Test() argument
624 …tmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxUint8Test()
631 float beta) in Simple3dSoftmaxUint8Test() argument
638 beta, in Simple3dSoftmaxUint8Test()
648 float beta) in Simple4dSoftmaxUint8Test() argument
652 …ftmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxUint8Test()
660 float beta) in SimpleSoftmaxFloat16Test() argument
662 …ftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxFloat16Test()
669 float beta) in Simple3dSoftmaxFloat16Test() argument
672 …oftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxFloat16Test()
680 float beta) in Simple4dSoftmaxFloat16Test() argument
683 …oftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxFloat16Test()
691 float beta) in SimpleSoftmaxUint16Test() argument
693 …tmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxUint16Test()
700 float beta) in Simple3dSoftmaxUint16Test() argument
703 …ftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxUint16Test()
711 float beta) in Simple4dSoftmaxUint16Test() argument
715 …ftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxUint16Test()
725 float beta) in CompareSoftmaxTest() argument
728 …loadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, beta); in CompareSoftmaxTest()
737 float beta) in CompareSoftmaxUint8Test() argument
740 …loadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, beta); in CompareSoftmaxUint8Test()