dopt.cpu

This module enables operation graphs to be evaluated using CPU kernels.

Modules

basic
module dopt.cpu.basic
Undocumented in source.
math
module dopt.cpu.math
Undocumented in source.
nnet
module dopt.cpu.nnet
Undocumented in source.
random
module dopt.cpu.random
Undocumented in source.

Members

Classes

CPUBuffer
class CPUBuffer
Undocumented in source.
CPUKernelDelegate
class CPUKernelDelegate

Convenience class that allows one to wrap a delegate and implement CPUKernel.

CPUPlan
class CPUPlan
Undocumented in source.

Functions

deregisterCPUKernel
void deregisterCPUKernel(string opName)

Deregisters the kernel associated with the specified operation.

evaluateCPU
DeviceBuffer[] evaluateCPU(Operation[] ops, DeviceBuffer[Operation] args)

Evaluates an several nodes from the operation graph using the CPU.

listAllCPUOperations
string[] listAllCPUOperations()

Provides a list of operations for which a CPUKernel has been registered.

registerCPUKernel
void registerCPUKernel(string opName, CPUKernel kernel)

Registers a kernel for the specified operation.

Interfaces

CPUKernel
interface CPUKernel

Common interface for all CPU kernels.

Meta

Authors

Henry Gouk