List of all items

Structs

• compiler_utils::GraphSearch
• compiler_utils::Looped
• compiler_utils::NewOp
• compiler_utils::SelectEdge
• compiler_utils::SelectOp
• compiler_utils::Timed
• compilers::ArithmeticElimination
• compilers::BatchMatMul2DCompiler
• compilers::BatchedMatMul2D
• compilers::CSE
• compilers::DepthFirst
• compilers::FusedUnary
• compilers::MatMul2D
• compilers::MatMul2DCompiler
• compilers::RemapDownstream
• compilers::RemoveSingleReductions
• compilers::RemoveUnusedNodes
• compilers::UnaryFusionCompiler
• compilers::UnarySequentialElimination
• graph::Graph
• graph::NodeIndex
• graph_tensor::GraphTensor
• nn::Repeated
• nn::activation::ReLU
• nn::activation::Sigmoid
• nn::activation::Swish
• nn::activation::Tanh
• nn::convolution::Conv1D
• nn::convolution::Conv2D
• nn::embedding::Embedding
• nn::linear::Linear
• nn::norm::LayerNorm
• nn::norm::RMSNorm
• nn::transformer::Transformer
• nn::transformer::attention::MultiHeadSelfAttention
• nn::transformer::decoder::TransformerDecoder
• nn::transformer::decoder::TransformerDecoderBlock
• nn::transformer::encoder::TransformerEncoderBlock
• op::Add
• op::Constant
• op::Contiguous
• op::Diff
• op::Exp2
• op::Function
• op::LessThan
• op::Log2
• op::MaxReduce
• op::Mod
• op::Mul
• op::Print
• op::Recip
• op::Sin
• op::Sqrt
• op::SumReduce
• prelude::Axes2
• prelude::Axes3
• prelude::Axes4
• prelude::Axes5
• prelude::Axes6
• prelude::Axis
• prelude::Const
• prelude::Dyn
• prelude::Graph
• prelude::GraphSearch
• prelude::GraphTensor
• prelude::Looped
• prelude::NewOp
• prelude::NodeIndex
• prelude::SafeTensorLoader
• prelude::SafeTensorSaver
• prelude::SelectEdge
• prelude::SelectOp
• prelude::Serializer
• prelude::ShapeTracker
• prelude::StateDictLoader
• prelude::StateDictSaver
• prelude::Tensor
• prelude::Timed
• prelude::bf16
• prelude::f16
• prelude::symbolic::GenericExpression
• prelude::tracker::ShapeTracker
• serialization::SafeTensorLoader
• serialization::SafeTensorSaver
• serialization::Serializer
• serialization::StateDictLoader
• serialization::StateDictSaver
• shape::Axes2
• shape::Axes3
• shape::Axes4
• shape::Axes5
• shape::Axes6
• shape::Axis
• shape::Const
• shape::Dyn
• shape::ShapeTracker
• shape::symbolic::GenericExpression
• shape::tracker::ShapeTracker
• tensor::Tensor

Enums

• graph::Dependency
• op::ConstantValue
• op::InputTensor
• prelude::Dependency
• prelude::ReshapeDim
• prelude::symbolic::Term
• shape::ReshapeDim
• shape::symbolic::Term

Traits

• compiler_utils::Compiler
• compiler_utils::JoinGraph
• compiler_utils::ToIds
• compiler_utils::ToIdsMut
• compiler_utils::TraitObjEq
• graph_tensor::MarkTensors
• graph_tensor::ToData
• module::InitModule
• module::Module
• op::Operator
• prelude::Array
• prelude::AssertSameNumel
• prelude::Axes
• prelude::BroadcastShapeTo
• prelude::BroadcastStridesTo
• prelude::Compiler
• prelude::ConstDim
• prelude::ConstShape
• prelude::Data
• prelude::Dimension
• prelude::HasAxes
• prelude::HasShape
• prelude::InitModule
• prelude::JoinGraph
• prelude::Loader
• prelude::MarkTensors
• prelude::Matmul
• prelude::Module
• prelude::PermuteShapeTo
• prelude::RangeToDim
• prelude::RealizeShapeTo
• prelude::ReduceShape
• prelude::ReduceShapeTo
• prelude::ReduceStridesTo
• prelude::Saver
• prelude::SerializeModule
• prelude::Shape
• prelude::SliceOfShape
• prelude::ToData
• prelude::ToIds
• prelude::ToIdsMut
• prelude::TraitObjEq
• prelude::matmul::Matmul
• serialization::Loader
• serialization::Saver
• serialization::SerializeModule
• shape::Array
• shape::AssertSameNumel
• shape::Axes
• shape::BroadcastShapeTo
• shape::BroadcastStridesTo
• shape::ConstDim
• shape::ConstShape
• shape::Dimension
• shape::HasAxes
• shape::HasShape
• shape::PermuteShapeTo
• shape::RangeToDim
• shape::RealizeShapeTo
• shape::ReduceShape
• shape::ReduceShapeTo
• shape::ReduceStridesTo
• shape::Shape
• shape::SliceOfShape
• tensor::Data

Macros

• compiler_utils::select_ty
• prelude::select_ty
• select_ty
• test_imports
• unwrap_cont

Derive Macros

• prelude::LuminalEqFalse
• prelude::LuminalEqTrue
• prelude::LuminalPrint

Functions

• compiler_utils::check_no_delete
• compiler_utils::display_graph
• compiler_utils::move_incoming_edge
• compiler_utils::move_outgoing_edge
• compiler_utils::move_references
• module::delete_inputs
• module::downstream
• module::state_dict
• module::state_set
• module::transfer_data
• module::transfer_data_same_graph
• op::get_vec_from_tensor
• op::get_vec_from_tensor_owned
• prelude::check_no_delete
• prelude::delete_inputs
• prelude::display_graph
• prelude::downstream
• prelude::move_incoming_edge
• prelude::move_outgoing_edge
• prelude::move_references
• prelude::resolve_local_dyn_dims
• prelude::state_dict
• prelude::state_set
• prelude::tracker::resolve_local_dyn_dims
• prelude::transfer_data
• prelude::transfer_data_same_graph
• shape::resolve_local_dyn_dims
• shape::tracker::resolve_local_dyn_dims
• tests::assert_close
• tests::assert_close_precision
• tests::assert_exact
• tests::random_vec
• tests::random_vec_rng
• tests::test_graphs::batch_matmul
• tests::test_graphs::feedforward
• tests::test_graphs::matmul
• tests::test_graphs::transformer

Type Aliases

• compilers::CPUCompiler
• compilers::GenericCompiler
• compilers::MatMulCompiler
• graph::MainGraph
• nn::transformer::encoder::TransformerEncoder
• prelude::MainGraph
• prelude::R0
• prelude::R1
• prelude::R2
• prelude::R3
• prelude::R4
• prelude::R5
• prelude::R6
• prelude::symbolic::BigExpression
• prelude::symbolic::Expression
• shape::R0
• shape::R1
• shape::R2
• shape::R3
• shape::R4
• shape::R5
• shape::R6
• shape::symbolic::BigExpression
• shape::symbolic::Expression