com.intel.analytics.bigdl.nn

mkldnn

package mkldnn

Visibility
  1. Public
  2. All

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 ReLU extends AbstractModule[Activity, Activity, Float] with MklDnnLayer with MklInt8Convertible

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

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

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

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

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

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

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

Value Members

  1. object AvgPooling extends Serializable

  2. object CAddTable extends Serializable

  3. object ConcatTable extends Serializable

  4. object Convolution

  5. object DnnGraph extends Serializable

  6. object Dropout extends Serializable

  7. object Identity extends Serializable

  8. object Input extends Serializable

  9. object JoinTable extends Serializable

  10. object LRN extends Serializable

  11. object Linear extends Serializable

  12. object MaxPooling extends Serializable

  13. object Output extends Serializable

  14. object Perf

  15. object Phase

  16. object ReLU extends Serializable

  17. object ReorderMemory extends Serializable

  18. object ResNet

  19. object SbnDnn

  20. object Scale

  21. object SelectTable extends Serializable

  22. object Sequential extends Serializable

  23. object SoftMax extends Serializable

  24. object SpatialBatchNormalization extends Serializable

  25. object SpatialConvolution extends Serializable

  26. package models

Ungrouped