org.apache.mxnet

CustomOpProp

Related Doc: package mxnet

abstract class CustomOpProp extends AnyRef

Base class for operator property class implemented in Scala. MXNET_CPU_WORKER_NTHREADS must be greater than 1 for custom op to work on CPU

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. CustomOpProp
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new CustomOpProp(needTopGrad: Boolean = false)

    needTopGrad

    : Boolean The default declareBackwardDependency function use this value to determine whether this operator needs gradient input for above.

Abstract Value Members

  1. abstract def createOperator(ctx: String, inShapes: Array[Array[Int]], inDtypes: Array[Int]): CustomOp

    Create an operator that carries out the real computation given the context, input shapes, and input data types.

Concrete Value Members

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

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

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

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

    Definition Classes
    Any
  5. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. def declareBackwardDependency(outGrad: Array[Int], inData: Array[Int], outData: Array[Int]): Array[Int]

    Declare dependencies of this operator for backward pass.

    Declare dependencies of this operator for backward pass.

    outGrad

    : array of Int ids of outGrad blobs.

    inData

    : array of Int ids of inData blobs.

    outData

    : array of Int ids of outData blobs.

    returns

    deps : array of Int ids of the needed blobs.

  7. final def eq(arg0: AnyRef): Boolean

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

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

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  10. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  11. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  12. def inferShape(inShape: Array[Shape]): (Array[Shape], Array[Shape], Array[Shape])

    inferShape interface.

    inferShape interface. override to create new operators

    inShape

    : array of Shape list of argument shapes in the same order as declared in listArguments().

    returns

    inShapes : array of Shape array of argument shapes. Can be modified from inShape. outShapes : array of Shape array of output shapes calculated from inShape, in the same order as declared in listOutputs(). auxShapes : Optional, array of Shape array of aux shapes calculated from inShape, in the same order as declared in listAuxiliaryStates().

  13. def inferType(inType: Array[DType.DType]): (Array[DType.DType], Array[DType.DType], Array[DType.DType])

    inferType interface.

    inferType interface. override to create new operators

    inType

    : array of DType list of argument types in the same order as declared in listArguments().

    returns

    inTypes : array of DType array of argument types. Can be modified from inType. outTypes : array of DType array of output types calculated from inType, in the same order as declared in listOutputs(). auxTypes : Optional, array of DType array of aux types calculated from inType, in the same order as declared in listAuxiliaryStates().

  14. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  15. var kwargs: Map[String, String]

    Attributes
    protected
  16. def listArguments(): Array[String]

    listArguments interface.

    listArguments interface. override to create new operators

    returns

    arguments : array of String list of argument blob names.

  17. def listAuxiliaryStates(): Array[String]

    listAuxiliaryStates interface.

    listAuxiliaryStates interface. override to create new operators

    returns

    auxs : array of String list of auxiliary state blob names.

  18. def listOutputs(): Array[String]

    listOutputs interface.

    listOutputs interface. override to create new operators

    returns

    outputs : array of String list of output blob names.

  19. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  20. final def notify(): Unit

    Definition Classes
    AnyRef
  21. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  22. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  23. def toString(): String

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

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped