Compute Library
 22.05
Conv3D.cpp File Reference

Go to the source code of this file.

Namespaces

 arm_compute
 Copyright (c) 2017-2022 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
 

Variable Documentation

◆ batch_dim

constexpr unsigned int batch_dim = 4u

◆ channel_dim

constexpr unsigned int channel_dim = 0u

◆ depth_dim

constexpr unsigned int depth_dim = 3u

◆ height_dim

constexpr unsigned int height_dim = 2u

◆ weights_CHin_dim

constexpr unsigned int weights_CHin_dim = 1u

Definition at line 42 of file Conv3D.cpp.

Referenced by arm_compute::test::validation::reference::conv3d().

◆ weights_CHout_dim

constexpr unsigned int weights_CHout_dim = 0u

◆ weights_depth_dim

constexpr unsigned int weights_depth_dim = 4u

◆ weights_height_dim

constexpr unsigned int weights_height_dim = 3u

◆ weights_width_dim

constexpr unsigned int weights_width_dim = 2u

◆ width_dim

constexpr unsigned int width_dim = 1u