mkldnn

Type Members

1. class AvgPooling extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

2. class CAddTable extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with MklInt8Convertible

3. class ConcatTable extends DynamicContainer[Activity, Activity, Float] with MklDnnContainer with MklInt8Convertible

4. class DnnGraph extends Graph[Float] with MklDnnLayer with MklInt8Convertible

5. class Dropout extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

6. case class HeapData(_shape: Array[Int], _layout: Int, _dataType: Int = ...) extends MemoryData with Product with Serializable

7. class Identity extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

   Identity just return the input to output.

8. class Input extends ReorderMemory

9. class JoinTable extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

10. class LRN extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

11. class Linear extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with Initializable with MklInt8Convertible

12. class MaxPooling extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

13. sealed trait MemoryData extends Serializable

14. trait MklDnnContainer extends DynamicContainer[Activity, Activity, Float] with MklDnnModule

    Helper utilities when integrating containers with MKL-DNN

15. trait MklDnnLayer extends AbstractModule[Activity, Activity, Float] with MklDnnModule

16. trait MklDnnModule extends MklDnnModuleHelper

    Helper utilities when integrating Module with MKL-DNN

17. trait MklDnnModuleHelper extends MemoryOwner

18. abstract class MklDnnNativeMemory extends Releasable

19. class MklDnnRuntime extends AnyRef

20. class MklMemoryAttr extends MklDnnNativeMemory

21. class MklMemoryDescInit extends MklDnnNativeMemory

22. class MklMemoryPostOps extends MklDnnNativeMemory

23. class MklMemoryPrimitive extends MklDnnNativeMemory

24. class MklMemoryPrimitiveDesc extends MklDnnNativeMemory

25. case class NativeData(_shape: Array[Int], _layout: Int, _dataType: Int = ...) extends MemoryData with Product with Serializable

26. class Output extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

    Convert output to user defined layout and appoint gradOutput layout

27. sealed class Phase extends AnyRef

28. class RNN extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with Initializable

29. class ReLU extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with MklInt8Convertible

30. class ReorderMemory extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with Releasable

31. case class ResNet50PerfParams(batchSize: Int = 16, iteration: Int = 50, training: Boolean = true, model: String = "vgg16") extends Product with Serializable

32. class SelectTable extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

    Creates a module that takes a table as input and outputs the element at index index (positive or negative).

33. class Sequential extends DynamicContainer[Activity, Activity, Float] with MklDnnContainer with MklInt8Convertible

34. class SoftMax extends AbstractModule[Activity, Activity, Float] with MklDnnLayer

35. class SpatialBatchNormalization extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with Initializable with MklInt8Convertible

36. class SpatialConvolution extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with Initializable with Serializable with MklInt8Convertible

    Applies a 2D convolution over an input image composed of several input planes.