Class

org.apache.mxnet.javaapi

ConvolutionParam

Related Doc: package javaapi

Permalink

class ConvolutionParam extends AnyRef

This Param Object is specifically used for Convolution

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. ConvolutionParam
  2. AnyRef
  3. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ConvolutionParam(data: NDArray, weight: NDArray, bias: NDArray, kernel: Shape, num_filter: Integer)

    Permalink

    data

    Input data to the ConvolutionOp.

    weight

    Weight matrix.

    bias

    Bias parameter.

    kernel

    Convolution kernel size: (w,), (h, w) or (d, h, w)

    num_filter

    Convolution filter(channel) number

Value Members

  1. final def !=(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Permalink
    Definition Classes
    Any
  5. def clone(): AnyRef

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean

    Permalink
    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

    Permalink
    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. def getBias(): NDArray

    Permalink
  10. final def getClass(): Class[_]

    Permalink
    Definition Classes
    AnyRef → Any
  11. def getCudnn_off(): Boolean

    Permalink
  12. def getCudnn_tune(): String

    Permalink
  13. def getData(): NDArray

    Permalink
  14. def getDilate(): Shape

    Permalink
  15. def getKernel(): Shape

    Permalink
  16. def getLayout(): String

    Permalink
  17. def getNo_bias(): Boolean

    Permalink
  18. def getNum_filter(): Integer

    Permalink
  19. def getNum_group(): Integer

    Permalink
  20. def getOut(): mxnet.NDArray

    Permalink
  21. def getPad(): Shape

    Permalink
  22. def getStride(): Shape

    Permalink
  23. def getWeight(): NDArray

    Permalink
  24. def getWorkspace(): Long

    Permalink
  25. def hashCode(): Int

    Permalink
    Definition Classes
    AnyRef → Any
  26. final def isInstanceOf[T0]: Boolean

    Permalink
    Definition Classes
    Any
  27. final def ne(arg0: AnyRef): Boolean

    Permalink
    Definition Classes
    AnyRef
  28. final def notify(): Unit

    Permalink
    Definition Classes
    AnyRef
  29. final def notifyAll(): Unit

    Permalink
    Definition Classes
    AnyRef
  30. def setCudnn_off(cudnn_off: Boolean): ConvolutionParam

    Permalink

    cudnn_off

    Turn off cudnn for this layer.

  31. def setCudnn_tune(cudnn_tune: String): ConvolutionParam

    Permalink

    cudnn_tune

    Whether to pick convolution algo by running performance test.

  32. def setDilate(dilate: Shape): ConvolutionParam

    Permalink

    dilate

    Convolution dilate: (w,), (h, w) or (d, h, w). Defaults to 1 for each dimension.

  33. def setLayout(layout: String): ConvolutionParam

    Permalink

    layout

    Set layout for input, output and weight. Empty for default layout: NCW for 1d, NCHW for 2d and NCDHW for 3d.NHWC and NDHWC are only supported on GPU.

  34. def setNo_bias(no_bias: Boolean): ConvolutionParam

    Permalink

    no_bias

    Whether to disable bias parameter.

  35. def setNum_group(num_group: Integer): ConvolutionParam

    Permalink

    num_group

    Number of group partitions.

  36. def setOut(out: NDArray): ConvolutionParam

    Permalink
  37. def setPad(pad: Shape): ConvolutionParam

    Permalink

    pad

    Zero pad for convolution: (w,), (h, w) or (d, h, w). Defaults to no padding.

  38. def setStride(stride: Shape): ConvolutionParam

    Permalink

    stride

    Convolution stride: (w,), (h, w) or (d, h, w). Defaults to 1 for each dimension.

  39. def setWorkspace(workspace: Long): ConvolutionParam

    Permalink

    workspace

    Maximum temporary workspace allowed (MB) in convolution.This parameter has two usages. When CUDNN is not used, it determines the effective batch size of the convolution kernel. When CUDNN is used, it controls the maximum temporary storage used for tuning the best CUDNN kernel when limited_workspace strategy is used.

  40. final def synchronized[T0](arg0: ⇒ T0): T0

    Permalink
    Definition Classes
    AnyRef
  41. def toString(): String

    Permalink
    Definition Classes
    AnyRef → Any
  42. final def wait(): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  43. final def wait(arg0: Long, arg1: Int): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  44. final def wait(arg0: Long): Unit

    Permalink
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped