skainet-backend-cpu/sk.ainet.context/DirectCpuExecutionContext

DirectCpuExecutionContext

class DirectCpuExecutionContext @JvmOverloads constructor(val executionStats: ExecutionStats = ExecutionStats(), val phase: Phase = Phase.EVAL, _hooks: ForwardHooks? = null, val tensorDataFactory: TensorDataFactory = DenseTensorDataFactory()) : ExecutionContext(source)

Constructors

DirectCpuExecutionContext

constructor(executionStats: ExecutionStats = ExecutionStats(), phase: Phase = Phase.EVAL, _hooks: ForwardHooks? = null, tensorDataFactory: TensorDataFactory = DenseTensorDataFactory())

Types

object Companion

Properties

open override val executionStats: ExecutionStats

open override val hooks: ForwardHooks?

open override val memoryInfo: MemoryInfo

open override val observers: ExecutionObserverRegistry

open override val ops: TensorOps

open override val phase: Phase

tensorDataFactory

open override val tensorDataFactory: TensorDataFactory