Field Detail

• context

  protected OpContext context

Constructor Detail

• MKLDNNBatchNormHelper

  public MKLDNNBatchNormHelper(DataType dataType)

Method Detail

• checkSupported

  public boolean checkSupported(double eps,
                                boolean fixedGammaBeta)

  Specified by:

  checkSupported in interface BatchNormalizationHelper

• backpropGradient

  public Pair<Gradient,INDArray> backpropGradient(INDArray input,
                                                  INDArray epsilon,
                                                  long[] shape,
                                                  INDArray gamma,
                                                  INDArray beta,
                                                  INDArray dGammaView,
                                                  INDArray dBetaView,
                                                  double eps,
                                                  LayerWorkspaceMgr workspaceMgr)

  Specified by:

  backpropGradient in interface BatchNormalizationHelper

• preOutput

  public INDArray preOutput(INDArray x,
                            boolean training,
                            long[] shape,
                            INDArray gamma,
                            INDArray beta,
                            INDArray mean,
                            INDArray var,
                            double decay,
                            double eps,
                            LayerWorkspaceMgr workspaceMgr)

  Specified by:

  preOutput in interface BatchNormalizationHelper

• getMeanCache

  public INDArray getMeanCache(DataType dataType)

  Specified by:

  getMeanCache in interface BatchNormalizationHelper

• getVarCache

  public INDArray getVarCache(DataType dataType)

  Specified by:

  getVarCache in interface BatchNormalizationHelper

• helperMemoryUse

  public Map<String,Long> helperMemoryUse()

  Description copied from interface: LayerHelper
  Return the currently allocated memory for the helper.
  (a) Excludes: any shared memory used by multiple helpers/layers
  (b) Excludes any temporary memory (c) Includes all memory that persists for longer than the helper method
  This is mainly used for debugging and reporting purposes. Returns a map:
  Key: The name of the type of memory
  Value: The amount of memory
  

  Specified by:

  helperMemoryUse in interface LayerHelper

  Returns:

  Map of memory, may be null if none is used.