mkldnn

Type Members

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

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

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

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

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

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

   Identity just return the input to output.

7. class Input extends ReorderMemory

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

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

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

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

12. sealed trait MemoryData extends Serializable

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

    Helper utilities when integrating containers with MKL-DNN

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

15. trait MklDnnModule extends MklDnnModuleHelper

    Helper utilities when integrating Module with MKL-DNN

16. trait MklDnnModuleHelper extends AnyRef

17. class MklDnnRuntime extends AnyRef

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

19. sealed class Phase extends AnyRef

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

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

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

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

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

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

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

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