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 AnyRef

18. class MklDnnRuntime extends AnyRef

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

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

    Convert output to user defined layout and appoint gradOutput layout

21. sealed class Phase extends AnyRef

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

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

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

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

26. 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).

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

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

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

30. 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.