Interactive Diagram Lab
Computational Graph & Tensor Dataflow Engine
Computational Graph & Tensor Dataflow Engine
Attention Is All You Need (Vaswani et al., 2017)
Computes semantic token-to-token similarity () and applies softmax weights over value representations
512 × 64
transpose→0 FLOPs
64 × 512
01Transpose Key Matrix
Transposes the [] Key matrix into [] for dot-product multiplication
Total Computation 68.16 MFLOPsOutput [512 × 64]
Core Architectural & Hardware Specification
Definition
A computational framework abstracting neural equations into a Directed Acyclic Graph (DAG), mapping tensors as dataflow edges and tracking autograd dependencies.
Core Mechanics
- •Decomposes math into primitive nodes (MatMul, Softmax, Norm) and tensor edges
- •Tracks backward paths by retaining forward activations to apply the chain rule
Limitations & Bottlenecks
- •Dynamic control flows limit ahead-of-time static JIT optimizations
Comparison
Unlike text equations, computational graphs explicitly expose operator independence, enabling hardware compilers (XLA, TensorRT) to execute automatic parallelization.