Compute Library
 21.08
Convolution3d.h File Reference

Go to the source code of this file.

Namespaces

 arm_compute
 Copyright (c) 2017-2021 Arm Limited.
 
 arm_compute::test
 
 arm_compute::test::convolution_3d
 
 arm_compute::test::convolution_3d::detail
 

Functions

bool is_valid_pixel (int i, int min, int max)
 
template<typename T , typename TW , typename TB , typename std::enable_if< validation::is_floating_point< T >::value &&validation::is_floating_point< TW >::value &&validation::is_floating_point< TB >::value, int >::type = 0>
void convolution3d (const SimpleTensor< T > &in, const SimpleTensor< TW > &weights, const SimpleTensor< TB > &bias, SimpleTensor< T > &out, int i_offset, int w_offset, int b_offset, int o_offset, int xi, int yi, int width_in, int height_in, int depth_in, int width_weights, int height_weights, int dilation_x=1, int dilation_y=1, int filter_id=0)