Graph Box

An AnvilBox subclass that wraps a GraphNode during graph construction (tracing). When a function is traced via trace_fn(), each intermediate tensor value is represented as a GraphBox. It also contains an associated GraphDescriptor in which the node "lives".

Usage

GraphBox(gnode, desc)

Arguments

gnode

(GraphNode)
The graph node – either a GraphValue or a GraphLiteral.

desc

(GraphDescriptor)
The descriptor of the graph being built.

Value

(GraphBox)

Extractors

• dtype()

• shape()

• ndims()

• ambiguous()

See also

AnvilBox, DebugBox, trace_fn(), jit()