utils

Type Members

1. class BlockManagerParameterSynchronizer[T] extends DistriParameterSynchronizer[T]

   

2. class DirectedGraph[T] extends Serializable

   

   Provides useful graph operations for a directed graph.

   Provides useful graph operations for a directed graph.

   The node connection in the graph has direction. This class stores a source node. Note that it doesn't maintain the topology. The topology of the graph is stored in the connection of the nodes.

   T

   Node element type

   Annotations

   @SerialVersionUID()

3. trait DistriParameterSynchronizer[T] extends AnyRef

   

4. sealed trait EngineType extends AnyRef

   

   define engine type trait

5. class LayerException extends RuntimeException

   

   Fine grained layer exception message

6. case class MultiShape(value: List[Shape]) extends Shape with Product with Serializable

   

7. class Node[T] extends Serializable

   

   Represent a node in a graph.

   Represent a node in a graph. The connections between nodes are directed.

   T

   element type

   Annotations

   @SerialVersionUID()

8. sealed trait OptimizerVersion extends AnyRef

   

   define optimizer version trait

9. class RandomGenerator extends AnyRef

   

   A mersenne twister based fake random number generator.

   A mersenne twister based fake random number generator. Please refer https://en.wikipedia.org/wiki/Mersenne_Twister. Note that it has its own state so it is not thread safe. So you should use RandomGenerator.RNG to get a thread local instance to use. That's thread-safe.

10. trait Shape extends AnyRef

    

11. case class SingleShape(value: List[Int]) extends Shape with Product with Serializable

    

12. case class SyncMeta[T](name: String, counter: Int, priority: Int, globalSize: Int, partitionToCount: Int, stateOfWorld: ConcurrentHashMap[Int, CompressedTensor[T]], aggregatedStateOfWorld: ConcurrentHashMap[Int, Tensor[T]], weights: Tensor[T] = null, grads: Tensor[T] = null) extends Product with Serializable

    

13. class Table extends Serializable with Activity

    

    Simulate the Table data structure in lua

14. class ThreadPool extends AnyRef

    

    A thread pool wrapper, provide some helper functions for multi-threading

    A thread pool wrapper, provide some helper functions for multi-threading

    TODO TreadPool will give 2-version of thread pool, one uses scala version (invokeAndWait), another is provided to Java (invokeAndWait2). The design is weird. We should refactor this class later.

15. sealed abstract class TorchObject extends AnyRef

    

    torch object type