dopt.core

This package contains the framework for constructing and executing operation graphs.

dopt.core.ops provides functions for constructing nodes in the operation graph.
dopt.core.grads provides functions for computing the derivatives of operations.
dopt.core.cpu contains a backend that executes operation graphs using the CPU.
dopt.core.cuda contains a backend that executes operation graphs using a CUDA enabled GPU.

Modules

grads module dopt.core.grads: Contains the automatic differentiation framework.
ops module dopt.core.ops: This package facilitates the construction of various nodes in the operation graph.
types module dopt.core.types

Members

Classes

Plan class Plan

Functions

compile Plan compile(Operation[] outputs): Compile an Operation graph into a reusable execution plan.
evaluate Buffer[] evaluate(Operation[] ops, Buffer[Operation] args = null): Evaluates a several nodes from the operation graph.
evaluate Buffer evaluate(Operation op, Buffer[Operation] args = null): Evaluates an operation graph with a single root node.

Meta

Source

See Source File

Authors

Henry Gouk

dopt modules