dragon/core¶

Classes¶

class CPUContext : The cpu device context.

class CUDAContext : The cuda device context.

class Graph : Graph to execute operators sequentially.

class MPSContext : The mps device context.

class Operator : The base operator class with context.

class OpSchema : Class to record the schema of operator.

class Tensor : The base tensor class.

class TypeMeta : The meta class for all types.

class UnifiedMemory : Memory to manage both the host and device data.

class Workspace : Sandbox to isolate the resources and computations.