ArmNN
 22.02
Functions
GatherTestImpl.cpp File Reference
#include "GatherTestImpl.hpp"
#include <ResolveType.hpp>
#include <armnnTestUtils/TensorCopyUtils.hpp>
#include <armnnTestUtils/WorkloadTestUtils.hpp>
#include <armnnTestUtils/TensorHelpers.hpp>

Go to the source code of this file.

Functions

LayerTestResult< float, 1 > Gather1dParamsFloat32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< armnn::Half, 1 > Gather1dParamsFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< uint8_t, 1 > Gather1dParamsUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int16_t, 1 > Gather1dParamsInt16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int32_t, 1 > Gather1dParamsInt32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< float, 2 > GatherMultiDimParamsFloat32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< armnn::Half, 2 > GatherMultiDimParamsFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< uint8_t, 2 > GatherMultiDimParamsUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int16_t, 2 > GatherMultiDimParamsInt16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int32_t, 2 > GatherMultiDimParamsInt32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< float, 4 > GatherMultiDimParamsMultiDimIndicesFloat32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< armnn::Half, 4 > GatherMultiDimParamsMultiDimIndicesFloat16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< uint8_t, 4 > GatherMultiDimParamsMultiDimIndicesUint8Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int16_t, 4 > GatherMultiDimParamsMultiDimIndicesInt16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 
LayerTestResult< int32_t, 4 > GatherMultiDimParamsMultiDimIndicesInt32Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
 

Function Documentation

◆ Gather1dParamsFloat16Test()

LayerTestResult<armnn::Half, 1> Gather1dParamsFloat16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 321 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

325 {
326  return GatherTestHelper<armnn::DataType::Float16>::Gather1dParamsTestImpl(
327  workloadFactory, memoryManager, tensorHandleFactory);
328 }

◆ Gather1dParamsFloat32Test()

LayerTestResult<float, 1> Gather1dParamsFloat32Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 312 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

316 {
317  return GatherTestHelper<armnn::DataType::Float32>::Gather1dParamsTestImpl(
318  workloadFactory, memoryManager, tensorHandleFactory);
319 }

◆ Gather1dParamsInt16Test()

LayerTestResult<int16_t, 1> Gather1dParamsInt16Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 339 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

343 {
344  return GatherTestHelper<armnn::DataType::QSymmS16>::Gather1dParamsTestImpl(
345  workloadFactory, memoryManager, tensorHandleFactory);
346 }

◆ Gather1dParamsInt32Test()

LayerTestResult<int32_t, 1> Gather1dParamsInt32Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 348 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

352 {
353  return GatherTestHelper<armnn::DataType::Signed32>::Gather1dParamsTestImpl(
354  workloadFactory, memoryManager, tensorHandleFactory);
355 }

◆ Gather1dParamsUint8Test()

LayerTestResult<uint8_t, 1> Gather1dParamsUint8Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 330 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

334 {
335  return GatherTestHelper<armnn::DataType::QAsymmU8>::Gather1dParamsTestImpl(
336  workloadFactory, memoryManager, tensorHandleFactory);
337 }

◆ GatherMultiDimParamsFloat16Test()

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

Definition at line 366 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

370 {
371  return GatherTestHelper<armnn::DataType::Float16>::GatherMultiDimParamsTestImpl(
372  workloadFactory, memoryManager, tensorHandleFactory);
373 }

◆ GatherMultiDimParamsFloat32Test()

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

Definition at line 357 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

361 {
362  return GatherTestHelper<armnn::DataType::Float32>::GatherMultiDimParamsTestImpl(
363  workloadFactory, memoryManager, tensorHandleFactory);
364 }

◆ GatherMultiDimParamsInt16Test()

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

Definition at line 384 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

388 {
389  return GatherTestHelper<armnn::DataType::QSymmS16>::GatherMultiDimParamsTestImpl(
390  workloadFactory, memoryManager, tensorHandleFactory);
391 }

◆ GatherMultiDimParamsInt32Test()

LayerTestResult<int32_t, 2> GatherMultiDimParamsInt32Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 393 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

397 {
398  return GatherTestHelper<armnn::DataType::Signed32>::GatherMultiDimParamsTestImpl(
399  workloadFactory, memoryManager, tensorHandleFactory);
400 }

◆ GatherMultiDimParamsMultiDimIndicesFloat16Test()

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

Definition at line 411 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

415 {
416  return GatherTestHelper<armnn::DataType::Float16>::GatherMultiDimParamsMultiDimIndicesTestImpl(
417  workloadFactory, memoryManager, tensorHandleFactory);
418 }

◆ GatherMultiDimParamsMultiDimIndicesFloat32Test()

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

Definition at line 402 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

406 {
407  return GatherTestHelper<armnn::DataType::Float32>::GatherMultiDimParamsMultiDimIndicesTestImpl(
408  workloadFactory, memoryManager, tensorHandleFactory);
409 }

◆ GatherMultiDimParamsMultiDimIndicesInt16Test()

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

Definition at line 429 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

433 {
434  return GatherTestHelper<armnn::DataType::QSymmS16>::GatherMultiDimParamsMultiDimIndicesTestImpl(
435  workloadFactory, memoryManager, tensorHandleFactory);
436 }

◆ GatherMultiDimParamsMultiDimIndicesInt32Test()

LayerTestResult<int32_t, 4> GatherMultiDimParamsMultiDimIndicesInt32Test ( armnn::IWorkloadFactory &  workloadFactory,
const armnn::IBackendInternal::IMemoryManagerSharedPtr &  memoryManager,
const armnn::ITensorHandleFactory &  tensorHandleFactory 
)

Definition at line 438 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

442 {
443  return GatherTestHelper<armnn::DataType::Signed32>::GatherMultiDimParamsMultiDimIndicesTestImpl(
444  workloadFactory, memoryManager, tensorHandleFactory);
445 }

◆ GatherMultiDimParamsMultiDimIndicesUint8Test()

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

Definition at line 420 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

424 {
425  return GatherTestHelper<armnn::DataType::QAsymmU8>::GatherMultiDimParamsMultiDimIndicesTestImpl(
426  workloadFactory, memoryManager, tensorHandleFactory);
427 }

◆ GatherMultiDimParamsUint8Test()

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

Definition at line 375 of file GatherTestImpl.cpp.

Referenced by TEST_SUITE().

379 {
380  return GatherTestHelper<armnn::DataType::QAsymmU8>::GatherMultiDimParamsTestImpl(
381  workloadFactory, memoryManager, tensorHandleFactory);
382 }