Class ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>

    • Field Detail

      • convolutionDim

        protected int convolutionDim
      • hasBias

        protected boolean hasBias
        If true (default): include bias parameters in the model. False: no bias.
      • kernelSize

        public int[] kernelSize
      • stride

        protected int[] stride
      • padding

        protected int[] padding
      • cudnnAlgoMode

        protected ConvolutionLayer.AlgoMode cudnnAlgoMode
        Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
      • cudnnAllowFallback

        protected boolean cudnnAllowFallback
        When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user. If false, the built-in (non-CuDNN) implementation for ConvolutionLayer will be used
    • Constructor Detail

      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride,
                                  int[] padding,
                                  int[] dilation,
                                  int dim)
      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride,
                                  int[] padding,
                                  int[] dilation)
      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride,
                                  int[] padding,
                                  int dim)
      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride,
                                  int[] padding)
      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride,
                                  int dim)
      • BaseConvBuilder

        protected BaseConvBuilder​(int[] kernelSize,
                                  int[] stride)
      • BaseConvBuilder

        protected BaseConvBuilder​(int dim,
                                  int... kernelSize)
      • BaseConvBuilder

        protected BaseConvBuilder​(int... kernelSize)
      • BaseConvBuilder

        protected BaseConvBuilder()
    • Method Detail

      • allowCausal

        protected abstract boolean allowCausal()
      • setConvolutionMode

        protected void setConvolutionMode​(ConvolutionMode convolutionMode)
      • hasBias

        public T hasBias​(boolean hasBias)
        If true (default): include bias parameters in the model. False: no bias.
        Parameters:
        hasBias - If true: include bias parameters in this model
      • convolutionMode

        public T convolutionMode​(ConvolutionMode convolutionMode)
        Set the convolution mode for the Convolution layer. See ConvolutionMode for more details
        Parameters:
        convolutionMode - Convolution mode for layer
      • kernelSize

        public T kernelSize​(int... kernelSize)
      • stride

        public T stride​(int... stride)
      • padding

        public T padding​(int... padding)
      • cudnnAlgoMode

        public T cudnnAlgoMode​(ConvolutionLayer.AlgoMode cudnnAlgoMode)
        Defaults to "PREFER_FASTEST", but "NO_WORKSPACE" uses less memory.
      • cudnnAllowFallback

        @Deprecated
        public T cudnnAllowFallback​(boolean allowFallback)
        When using CuDNN and an error is encountered, should fallback to the non-CuDNN implementatation be allowed? If set to false, an exception in CuDNN will be propagated back to the user. If true, the built-in (non-CuDNN) implementation for ConvolutionLayer will be used
        Parameters:
        allowFallback - Whether fallback to non-CuDNN implementation should be used
      • helperAllowFallback

        public T helperAllowFallback​(boolean allowFallback)
        When using CuDNN or MKLDNN and an error is encountered, should fallback to the non-helper implementation be allowed? If set to false, an exception in the helper will be propagated back to the user. If true, the built-in (non-MKL/CuDNN) implementation for ConvolutionLayer will be used
        Parameters:
        allowFallback - Whether fallback to non-CuDNN implementation should be used