Module graphs

Crucial structs, enums, functions and types to create computation graphs.

Modules

util

Structs

Context

A structure that stores a pointer to a computation context that contains related computation graphs.

Graph

A structure that stores a pointer to a computation graph, where every node corresponds to an operation.

Node

A structure that stores a pointer to a computation graph node that corresponds to an operation.

ShardConfig

Shard config contains the parameters of the Sharding operation, namely:

Enums

JoinType

SliceElement

This enum represents different types of slice elements that are used to create indexing slices (see Slice and Graph::get_slice).

Functions

contexts_deep_equal

Check that two given contexts contain the same data, i.e. graphs, nodes, names, parameters.

create_context

Creates an empty computation context.

Type Aliases

Slice

Slice type denotes an indexing slice (see NumPy slicing).