SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Public Member Functions | Protected Types | List of all members
smaug::DepthwiseConvolutionOp< Backend > Class Template Reference

Implements the depthwise convolution operator. More...

#include <depthwise_convolution_op.h>

Detailed Description

template<typename Backend>
class smaug::DepthwiseConvolutionOp< Backend >

Implements the depthwise convolution operator.

Template Parameters
BackendThe Backend specialization of this Operator.

Definition at line 35 of file backend.h.

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
 

The documentation for this class was generated from the following files: