Compute Library
18.02
|
Stripped down version of AlexNet as not all kernel sizes and strides are supported. More...
#include <DirectConvolutionLayerDataset.h>
Public Member Functions | |
DirectConvolutionLayerDataset () | |
![]() | |
iterator | begin () const |
int | size () const |
void | add_config (TensorShape src, TensorShape weights, TensorShape biases, TensorShape dst, PadStrideInfo info) |
Additional Inherited Members | |
![]() | |
using | type = std::tuple< TensorShape, TensorShape, TensorShape, TensorShape, PadStrideInfo > |
Stripped down version of AlexNet as not all kernel sizes and strides are supported.
Definition at line 41 of file DirectConvolutionLayerDataset.h.
|
inline |
Definition at line 44 of file DirectConvolutionLayerDataset.h.
References ConvolutionLayerDataset::add_config(), arm_compute::FLOOR, and arm_compute::U.