colStride (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
computeOutputDim(int inputDim, int weightDim, int stride, PaddingType pad) const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inlineprotected |
computeOutputDim(int inputDim, int weightDim, int stride, int padding) const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inlineprotected |
ConvolutionOp(const std::string &name, Workspace *workspace) (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
createAllTensors() override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
createOutputTensors() (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
createWeightsTensors() | smaug::ConvolutionOp< SmvBackend > | inline |
getColStride() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getInputPadding() const | smaug::ConvolutionOp< SmvBackend > | inline |
getNumOfmaps() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getNumParameters() const override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getPadding() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getParameterizableInputs() override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getRowStride() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getWeightCols() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
getWeightRows() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
inferOutputShape() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inlinevirtual |
inferWeightsShape() const (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inlinevirtual |
Inputs enum value (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | |
invokeSystolicArrayKernel(unsigned accelId, float16 *inputs, float16 *weights, float16 *outputs, int inputsDims[4], int weightsDims[4], int outputsDims[4], int inputsPad, int weightsPad, int outputPad, int inputHaloPad[4], int stride, int ifmapStart, int kernStart, bool accumulate, bool readInputs, bool readWeights, bool sendResults, ActivationInfo *actInfo) (defined in smaug::SmvConvolutionOp) | smaug::SmvConvolutionOp | protected |
isSamplingSupported() const override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
Kernels enum value (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | |
kNumInputs enum value (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | |
kNumOutputs enum value (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | |
numOfmaps (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
Outputs enum value (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | |
paddingType (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
rowStride (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
run() override (defined in smaug::SmvConvolutionOp) | smaug::SmvConvolutionOp | |
runNHWC(TiledTensor &inputs, TiledTensor &weights, TiledTensor &outputs) | smaug::SmvConvolutionOp | protected |
sampling (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
setPadding(PaddingType padding) (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
setSamplingInfo(const SamplingInfo &_sampling) override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
setStride(int _rowStride, int _colStride) (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
setWeightDims(int _weightRows, int _weightCols, int _numOfmaps) (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
smv::conv::TilingOptimizer (defined in smaug::SmvConvolutionOp) | smaug::SmvConvolutionOp | friend |
tile() override (defined in smaug::SmvConvolutionOp) | smaug::SmvConvolutionOp | |
tiledTensors (defined in smaug::SmvConvolutionOp) | smaug::SmvConvolutionOp | protected |
validate() override (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | inline |
weightCols (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
weightRows (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |
weightsName (defined in smaug::ConvolutionOp< SmvBackend >) | smaug::ConvolutionOp< SmvBackend > | protected |