ArmNN
 21.11
Functions
SoftmaxTestImpl.hpp File Reference
#include "LayerTestResult.hpp"
#include <Half.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <backendsCommon/WorkloadFactory.hpp>

Go to the source code of this file.

Functions

LayerTestResult< float, 2 > SimpleSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< float, 2 > SimpleAxisSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta, int axis)
 
LayerTestResult< float, 3 > Simple3dSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< float, 3 > Simple3dAxisSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta, int axis)
 
LayerTestResult< float, 4 > Simple4dSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< float, 4 > Simple4dAxisSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta, int axis)
 
LayerTestResult< uint8_t, 2 > SimpleSoftmaxUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< uint8_t, 3 > Simple3dSoftmaxUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< uint8_t, 4 > Simple4dSoftmaxUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< armnn::Half, 2 > SimpleSoftmaxFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< armnn::Half, 3 > Simple3dSoftmaxFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< armnn::Half, 4 > Simple4dSoftmaxFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< int16_t, 2 > SimpleSoftmaxUint16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< int16_t, 3 > Simple3dSoftmaxUint16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< int16_t, 4 > Simple4dSoftmaxUint16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory, float beta)
 
LayerTestResult< float, 2 > CompareSoftmaxTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, armnn::IWorkloadFactory &refWorkloadFactory, const armnn::ITensorHandleFactory &tensorHandleFactory, const armnn::ITensorHandleFactory &refTensorHandleFactory, float beta)
 
LayerTestResult< uint8_t, 2 > CompareSoftmaxUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, armnn::IWorkloadFactory &refWorkloadFactory, const armnn::ITensorHandleFactory &tensorHandleFactory, const armnn::ITensorHandleFactory &refTensorHandleFactory, float beta)
 

Function Documentation

◆ CompareSoftmaxTest()

LayerTestResult<float, 2> CompareSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
armnn::IWorkloadFactory &  refWorkloadFactory,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
const armnn::ITensorHandleFactory &  refTensorHandleFactory,
float  beta 
)

Definition at line 717 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

724 {
725  return CompareSoftmaxTestImpl<armnn::DataType::Float32>(
726  workloadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, beta);
727 }

◆ CompareSoftmaxUint8Test()

LayerTestResult<uint8_t, 2> CompareSoftmaxUint8Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
armnn::IWorkloadFactory &  refWorkloadFactory,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
const armnn::ITensorHandleFactory &  refTensorHandleFactory,
float  beta 
)

Definition at line 729 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

736 {
737  return CompareSoftmaxTestImpl<armnn::DataType::QAsymmU8>(
738  workloadFactory, memoryManager, refWorkloadFactory, tensorHandleFactory, refTensorHandleFactory, beta);
739 }

◆ Simple3dAxisSoftmaxTest()

LayerTestResult<float, 3> Simple3dAxisSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta,
int  axis 
)

Definition at line 344 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

350 {
351  armnn::TensorShape inputShape;
352  std::vector<float> inputData;
353  std::vector<float> outputData;
354  switch (axis)
355  {
356  case -3:
357  case 0:
358  {
359  inputShape = {5, 2, 2};
360 
361  inputData =
362  {
363  17.0f, -1.0f, 17.0f, -1.0f, 16.0f, -2.0f, 16.0f, -2.0f, 15.0f, -3.0f,
364 
365  15.0f, -3.0f, 14.0f, -4.0f, 14.0f, -4.0f, 1.0f, -17.0f, 1.0f, -17.0f
366  };
367 
368  outputData =
369  {
370  0.643914213228014f, 0.643914213228014f, 0.643914213228014f, 0.643914213228014f,
371  0.236882800924671f,
372  0.236882800924671f, 0.236882800924671f, 0.236882800924671f, 0.087144312427294f,
373  0.087144312427294f,
374 
375  0.087144312427294f, 0.087144312427294f, 0.032058600957022f, 0.032058600957022f,
376  0.032058600957022f,
377  0.032058600957022f, 7.246299848982885e-08f, 7.246299848982885e-08f, 7.246299848982885e-08f,
378  7.246299848982885e-08f
379  };
380  break;
381  }
382  case -2:
383  case 1:
384  {
385  inputShape = {2, 5, 2};
386 
387  inputData =
388  {
389  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f,
390 
391  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f
392  };
393 
394  outputData =
395  {
396  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
397  0.087144312427294f,
398  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
399  7.246299848982885e-08f,
400 
401  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
402  0.087144312427294f,
403  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
404  7.246299848982885e-08f
405  };
406  break;
407  }
408  case -1:
409  case 2:
410  {
411  inputShape = {2, 2, 5};
412 
413  inputData =
414  {
415  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f,
416  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f
417  };
418 
419  outputData =
420  {
421  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
422  7.246299848982885e-08f,
423  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
424  7.246299848982885e-08f,
425 
426  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
427  7.246299848982885e-08f,
428  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
429  7.246299848982885e-08f
430  };
431  break;
432  }
433  }
434 
435  return Simple3dSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta,
436  inputShape, outputData, inputData, axis);
437 }

◆ Simple3dSoftmaxFloat16Test()

LayerTestResult<armnn::Half,3> Simple3dSoftmaxFloat16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 663 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

668 {
669  Simple3dSoftmaxOutputData data;
670  return Simple3dSoftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta,
671  data.inputShape, data.outputData, data.inputData);
672 }

◆ Simple3dSoftmaxTest()

LayerTestResult<float, 3> Simple3dSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 333 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

338 {
339  Simple3dSoftmaxOutputData data;
340  return Simple3dSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta,
341  data.inputShape, data.outputData, data.inputData);
342 }

◆ Simple3dSoftmaxUint16Test()

LayerTestResult<int16_t,3> Simple3dSoftmaxUint16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 694 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

699 {
700  Simple3dSoftmaxOutputData data;
701  return Simple3dSoftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta,
702  data.inputShape, data.outputData, data.inputData);
703 }

◆ Simple3dSoftmaxUint8Test()

LayerTestResult<uint8_t,3> Simple3dSoftmaxUint8Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 625 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

630 {
631  Simple3dSoftmaxOutputData data;
632  return Simple3dSoftmaxTestImpl<armnn::DataType::QAsymmU8>(
633  workloadFactory,
634  memoryManager,
635  tensorHandleFactory,
636  beta,
637  data.inputShape,
638  data.outputData,
639  data.inputData);
640 }

◆ Simple4dAxisSoftmaxTest()

LayerTestResult<float, 4> Simple4dAxisSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta,
int  axis 
)

Definition at line 450 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

456 {
457  armnn::TensorShape inputShape;
458  std::vector<float> inputData;
459  std::vector<float> outputData;
460  switch (axis)
461  {
462  case -4:
463  case 0:
464  {
465  inputShape = {5, 2, 2, 2};
466 
467  inputData =
468  {
469  17.0f, -1.0f, 17.0f, -1.0f, 17.0f, -1.0f, 17.0f, -1.0f, 16.0f, -2.0f,
470  16.0f, -2.0f, 16.0f, -2.0f, 16.0f, -2.0f, 15.0f, -3.0f, 15.0f, -3.0f,
471  15.0f, -3.0f, 15.0f, -3.0f, 14.0f, -4.0f, 14.0f, -4.0f, 14.0f, -4.0f,
472  14.0f, -4.0f, 1.0f, -17.0f, 1.0f, -17.0f, 1.0f, -17.0f, 1.0f, -17.0f
473  };
474 
475  outputData =
476  {
477  0.643914213228014f, 0.643914213228014f, 0.643914213228014f, 0.643914213228014f,
478  0.643914213228014f,
479  0.643914213228014f, 0.643914213228014f, 0.643914213228014f, 0.236882800924671f,
480  0.236882800924671f,
481  0.236882800924671f, 0.236882800924671f, 0.236882800924671f, 0.236882800924671f,
482  0.236882800924671f,
483  0.236882800924671f, 0.087144312427294f, 0.087144312427294f, 0.087144312427294f,
484  0.087144312427294f,
485 
486  0.087144312427294f, 0.087144312427294f, 0.087144312427294f, 0.087144312427294f,
487  0.032058600957022f,
488  0.032058600957022f, 0.032058600957022f, 0.032058600957022f, 0.032058600957022f,
489  0.032058600957022f,
490  0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f, 7.246299848982885e-08f,
491  7.246299848982885e-08f,
492  7.246299848982885e-08f, 7.246299848982885e-08f, 7.246299848982885e-08f,
493  7.246299848982885e-08f, 7.246299848982885e-08f
494  };
495  break;
496  }
497  case -3:
498  case 1:
499  {
500  inputShape = {2, 5, 2, 2};
501 
502  inputData =
503  {
504  17.0f, -1.0f, 17.0f, -1.0f, 16.0f, -2.0f, 16.0f, -2.0f, 15.0f, -3.0f,
505  15.0f, -3.0f, 14.0f, -4.0f, 14.0f, -4.0f, 1.0f, -17.0f, 1.0f, -17.0f,
506  17.0f, -1.0f, 17.0f, -1.0f, 16.0f, -2.0f, 16.0f, -2.0f, 15.0f, -3.0f,
507  15.0f, -3.0f, 14.0f, -4.0f, 14.0f, -4.0f, 1.0f, -17.0f, 1.0f, -17.0f
508  };
509 
510  outputData =
511  {
512  0.643914213228014f, 0.643914213228014f, 0.643914213228014f, 0.643914213228014f,
513  0.236882800924671f,
514  0.236882800924671f, 0.236882800924671f, 0.236882800924671f, 0.087144312427294f,
515  0.087144312427294f,
516  0.087144312427294f, 0.087144312427294f, 0.032058600957022f, 0.032058600957022f,
517  0.032058600957022f,
518  0.032058600957022f, 7.246299848982885e-08f, 7.246299848982885e-08f, 7.246299848982885e-08f,
519  7.246299848982885e-08f,
520 
521 
522  0.643914213228014f, 0.643914213228014f, 0.643914213228014f, 0.643914213228014f,
523  0.236882800924671f,
524  0.236882800924671f, 0.236882800924671f, 0.236882800924671f, 0.087144312427294f,
525  0.087144312427294f,
526  0.087144312427294f, 0.087144312427294f, 0.032058600957022f, 0.032058600957022f,
527  0.032058600957022f,
528  0.032058600957022f, 7.246299848982885e-08f, 7.246299848982885e-08f, 7.246299848982885e-08f,
529  7.246299848982885e-08f
530  };
531  break;
532  }
533  case -2:
534  case 2:
535  {
536  inputShape = {2, 2, 5, 2};
537 
538  inputData =
539  {
540  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f,
541  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f,
542  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f,
543  17.0f, -1.0f, 16.0f, -2.0f, 15.0f, -3.0f, 14.0f, -4.0f, 1.0f, -17.0f
544  };
545 
546  outputData =
547  {
548  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
549  0.087144312427294f,
550  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
551  7.246299848982885e-08f,
552  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
553  0.087144312427294f,
554  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
555  7.246299848982885e-08f,
556 
557  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
558  0.087144312427294f,
559  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
560  7.246299848982885e-08f,
561  0.643914213228014f, 0.643914213228014f, 0.236882800924671f, 0.236882800924671f,
562  0.087144312427294f,
563  0.087144312427294f, 0.032058600957022f, 0.032058600957022f, 7.246299848982885e-08f,
564  7.246299848982885e-08f
565  };
566  break;
567  }
568  case -1:
569  case 3:
570  {
571  inputShape = {2, 2, 2, 5};
572 
573  inputData =
574  {
575  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f,
576  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f,
577  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f,
578  17.0f, 16.0f, 15.0f, 14.0f, 1.0f, -1.0f, -2.0f, -3.0f, -4.0f, -17.0f
579  };
580 
581  outputData =
582  {
583  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
584  7.246299848982885e-08f,
585  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
586  7.246299848982885e-08f,
587  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
588  7.246299848982885e-08f,
589  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
590  7.246299848982885e-08f,
591 
592  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
593  7.246299848982885e-08f,
594  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
595  7.246299848982885e-08f,
596  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
597  7.246299848982885e-08f,
598  0.643914213228014f, 0.236882800924671f, 0.087144312427294f, 0.032058600957022f,
599  7.246299848982885e-08f
600  };
601  break;
602  }
603  }
604 
605  return Simple4dSoftmaxTestImpl<armnn::DataType::Float32>(
606  workloadFactory,
607  memoryManager,
608  tensorHandleFactory,
609  beta,
610  inputShape,
611  outputData,
612  inputData,
613  axis);
614 }

◆ Simple4dSoftmaxFloat16Test()

LayerTestResult<armnn::Half,4> Simple4dSoftmaxFloat16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 674 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

679 {
680  Simple4dSoftmaxData data;
681  return Simple4dSoftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta,
682  data.inputShape, data.outputData, data.inputData);
683 }

◆ Simple4dSoftmaxTest()

LayerTestResult<float, 4> Simple4dSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 439 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

444 {
445  Simple4dSoftmaxData data;
446  return Simple4dSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory,
447  beta, data.inputShape, data.outputData, data.inputData);
448 }

◆ Simple4dSoftmaxUint16Test()

LayerTestResult<int16_t,4> Simple4dSoftmaxUint16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 705 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

710 {
711  Simple4dSoftmaxData data;
712 
713  return Simple4dSoftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta,
714  data.inputShape, data.outputData, data.inputData);
715 }

◆ Simple4dSoftmaxUint8Test()

LayerTestResult<uint8_t,4> Simple4dSoftmaxUint8Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 642 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

647 {
648  Simple4dSoftmaxData data;
649 
650  return Simple4dSoftmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta,
651  data.inputShape, data.outputData, data.inputData);
652 }

◆ SimpleAxisSoftmaxTest()

LayerTestResult<float, 2> SimpleAxisSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta,
int  axis 
)

Definition at line 322 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

328 {
329  return SimpleSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager,
330  tensorHandleFactory, beta, axis);
331 }

◆ SimpleSoftmaxFloat16Test()

LayerTestResult<armnn::Half,2> SimpleSoftmaxFloat16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 654 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

659 {
660  return SimpleSoftmaxTestImpl<armnn::DataType::Float16>(workloadFactory, memoryManager, tensorHandleFactory, beta);
661 }

◆ SimpleSoftmaxTest()

LayerTestResult<float, 2> SimpleSoftmaxTest ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 313 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

318 {
319  return SimpleSoftmaxTestImpl<armnn::DataType::Float32>(workloadFactory, memoryManager, tensorHandleFactory, beta);
320 }

◆ SimpleSoftmaxUint16Test()

LayerTestResult<int16_t,2> SimpleSoftmaxUint16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 685 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

690 {
691  return SimpleSoftmaxTestImpl<armnn::DataType::QSymmS16>(workloadFactory, memoryManager, tensorHandleFactory, beta);
692 }

◆ SimpleSoftmaxUint8Test()

LayerTestResult<uint8_t, 2> SimpleSoftmaxUint8Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory,
float  beta 
)

Definition at line 616 of file SoftmaxTestImpl.cpp.

Referenced by TEST_SUITE().

621 {
622  return SimpleSoftmaxTestImpl<armnn::DataType::QAsymmU8>(workloadFactory, memoryManager, tensorHandleFactory, beta);
623 }