• Home
  • Raw
  • Download

Lines Matching refs:tensorHandleFactory

62     const armnn::ITensorHandleFactory& tensorHandleFactory,  in SimpleSoftmaxBaseTestImpl()  argument
90 …std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputT… in SimpleSoftmaxBaseTestImpl()
91 …std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(output… in SimpleSoftmaxBaseTestImpl()
123 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxTestImpl() argument
145 … SimpleSoftmaxBaseTestImpl<ArmnnType, 2>(workloadFactory, memoryManager, tensorHandleFactory, beta, in SimpleSoftmaxTestImpl()
153 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxTestImpl() argument
201 … SimpleSoftmaxBaseTestImpl<ArmnnType, 2>(workloadFactory, memoryManager, tensorHandleFactory, beta, in SimpleSoftmaxTestImpl()
209 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dSoftmaxTestImpl() argument
216 … SimpleSoftmaxBaseTestImpl<ArmnnType, 3>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxTestImpl()
224 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dSoftmaxTestImpl() argument
232 … SimpleSoftmaxBaseTestImpl<ArmnnType, 4>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxTestImpl()
241 const armnn::ITensorHandleFactory& tensorHandleFactory, in CompareSoftmaxTestImpl() argument
266 …std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputT… in CompareSoftmaxTestImpl()
267 …std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(output… in CompareSoftmaxTestImpl()
313 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxTest() argument
316 …ftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxTest()
322 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleAxisSoftmaxTest() argument
327 tensorHandleFactory, beta, axis); in SimpleAxisSoftmaxTest()
333 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dSoftmaxTest() argument
337 …oftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxTest()
344 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dAxisSoftmaxTest() argument
432 …oftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dAxisSoftmaxTest()
439 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dSoftmaxTest() argument
443 …ple4dSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, in Simple4dSoftmaxTest()
450 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dAxisSoftmaxTest() argument
605 tensorHandleFactory, in Simple4dAxisSoftmaxTest()
616 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxUint8Test() argument
619 …tmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxUint8Test()
625 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dSoftmaxUint8Test() argument
632 tensorHandleFactory, in Simple3dSoftmaxUint8Test()
642 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dSoftmaxUint8Test() argument
647 …ftmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxUint8Test()
654 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxFloat16Test() argument
657 …ftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxFloat16Test()
663 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dSoftmaxFloat16Test() argument
667 …oftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxFloat16Test()
674 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dSoftmaxFloat16Test() argument
678 …oftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxFloat16Test()
685 const armnn::ITensorHandleFactory& tensorHandleFactory, in SimpleSoftmaxUint16Test() argument
688 …tmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta); in SimpleSoftmaxUint16Test()
694 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple3dSoftmaxUint16Test() argument
698 …ftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple3dSoftmaxUint16Test()
705 const armnn::ITensorHandleFactory& tensorHandleFactory, in Simple4dSoftmaxUint16Test() argument
710 …ftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta, in Simple4dSoftmaxUint16Test()
718 const armnn::ITensorHandleFactory& tensorHandleFactory, in CompareSoftmaxTest() argument
723 …workloadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, b… in CompareSoftmaxTest()
730 const armnn::ITensorHandleFactory& tensorHandleFactory, in CompareSoftmaxUint8Test() argument
735 …workloadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, b… in CompareSoftmaxUint8Test()