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. case class HeapData(_shape: Array[Int], _layout: Int) extends MemoryData with Product with Serializable

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

   Identity just return the input to output.

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

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

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

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

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

11. sealed trait MemoryData extends Serializable

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

    Helper utilities when integrating containers with MKL-DNN

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

14. trait MklDnnModule extends MklDnnModuleHelper

    Helper utilities when integrating Module with MKL-DNN

15. trait MklDnnModuleHelper extends AnyRef

16. class MklDnnRuntime extends AnyRef

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

18. sealed class Phase extends AnyRef

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

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

21. case class ResNet50PerfParams(batchSize: Int = 16, iteration: Int = 50, training: Boolean = true) extends Product with Serializable

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

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

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

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

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