24.02.1
|
#include "Conv3D.h"
#include "arm_compute/core/utils/misc/ShapeCalculator.h"
#include "arm_compute/core/utils/quantization/AsymmHelpers.h"
#include "support/AclRequires.h"
#include "tests/validation/reference/UtilsQuantizedAsymm.h"
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2024 Arm Limited. | |
arm_compute::test | |
arm_compute::test::validation | |
arm_compute::test::validation::reference | |
Functions | |
template<typename T , typename TB > | |
SimpleTensor< T > | conv3d (const SimpleTensor< T > &src, const SimpleTensor< T > &weights, const SimpleTensor< TB > &bias, SimpleTensor< T > &dst, const Conv3dInfo &conv3d_info) |
template SimpleTensor< float > | conv3d (const SimpleTensor< float > &src, const SimpleTensor< float > &weights, const SimpleTensor< float > &bias, SimpleTensor< float > &dst, const Conv3dInfo &conv3d_info) |
template SimpleTensor< half > | conv3d (const SimpleTensor< half > &src, const SimpleTensor< half > &weights, const SimpleTensor< half > &bias, SimpleTensor< half > &dst, const Conv3dInfo &conv3d_info) |
template SimpleTensor< uint8_t > | conv3d (const SimpleTensor< uint8_t > &src, const SimpleTensor< uint8_t > &weights, const SimpleTensor< int32_t > &bias, SimpleTensor< uint8_t > &dst, const Conv3dInfo &conv3d_info) |
template SimpleTensor< int8_t > | conv3d (const SimpleTensor< int8_t > &src, const SimpleTensor< int8_t > &weights, const SimpleTensor< int32_t > &bias, SimpleTensor< int8_t > &dst, const Conv3dInfo &conv3d_info) |
Variables | |
constexpr unsigned int | batch_dim = 4u |
constexpr unsigned int | depth_dim = 3u |
constexpr unsigned int | height_dim = 2u |
constexpr unsigned int | width_dim = 1u |
constexpr unsigned int | channel_dim = 0u |
constexpr unsigned int | weights_depth_dim = 4u |
constexpr unsigned int | weights_height_dim = 3u |
constexpr unsigned int | weights_width_dim = 2u |
constexpr unsigned int | weights_CHin_dim = 1u |
constexpr unsigned int | weights_CHout_dim = 0u |
|
constexpr |
Definition at line 32 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 36 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 33 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 34 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 42 of file Conv3D.cpp.
Referenced by arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 43 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().
|
constexpr |
Definition at line 39 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape().
|
constexpr |
Definition at line 40 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape().
|
constexpr |
Definition at line 41 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape().
|
constexpr |
Definition at line 35 of file Conv3D.cpp.
Referenced by arm_compute::misc::shape_calculator::compute_conv3d_shape(), and arm_compute::test::validation::reference::conv3d().