dopt.cpu

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

Members

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

deregisterCPUKernel void deregisterCPUKernel(string opName): Deregisters the kernel associated with the specified operation.
evaluateCPU Buffer[] evaluateCPU(Operation[] ops, Buffer[Operation] args = null): 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.