SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Implements the depthwise convolution operator. More...
#include <depthwise_convolution_op.h>
Implements the depthwise convolution operator.
Backend | The Backend specialization of this Operator. |
Public Member Functions | |
DepthwiseConvolutionOp (const std::string &name, Workspace *workspace) | |
void | run () override |
TensorShape | inferOutputShape () const override |
TensorShape | inferWeightsShape () const override |
void | run () |
Protected Types | |
typedef ConvolutionOp< Backend > | Parent |