List of all items
Structs
- BackendsManifest
- CompilePipeline
- CompileResult
- CompiledGraph
- DeviceBenchResult
- DeviceCandidate
- DeviceFallbackError
- DevicePolicy
- DeviceRouter
- Element
- FlexibleSession
- FusionOptions
- FusionReport
- Graph
- GraphDevices
- GraphModule
- HirModule
- LirModule
- MirModule
- MissedFusion
- Node
- NodeId
- NodeOrigin
- ParseDeviceError
- PipelineInspect
- Session
- Shape
- Tick
- prelude::BackendsManifest
- prelude::CompiledGraph
- prelude::DeviceBenchResult
- prelude::DeviceCandidate
- prelude::DeviceFallbackError
- prelude::DevicePolicy
- prelude::DeviceRouter
- prelude::Element
- prelude::FlexibleSession
- prelude::Graph
- prelude::GraphDevices
- prelude::GraphModule
- prelude::Node
- prelude::NodeId
- prelude::ParseDeviceError
- prelude::RlxEnv
- prelude::RuntimeOverrides
- prelude::Session
- prelude::Shape
- prelude::Tick
- prelude::env::RlxEnv
- prelude::env::RuntimeOverrides
- quant::QuantMap
Enums
- DType
- Device
- DevicePickStrategy
- FusionPolicy
- FusionTarget
- GraphStage
- HirOp
- MissReason
- Op
- OpKind
- Precision
- PrecisionPolicy
- QuantScheme
- ops::Activation
- ops::BinaryOp
- ops::ChainOperand
- ops::ChainStep
- ops::CmpOp
- ops::MaskKind
- prelude::Activation
- prelude::BinaryOp
- prelude::CmpOp
- prelude::DType
- prelude::Device
- prelude::DevicePickStrategy
- prelude::GraphStage
- prelude::MaskKind
- prelude::Op
- prelude::OpKind
- prelude::Precision
- prelude::PrecisionPolicy
- prelude::QuantScheme
- quant::QuantScheme
Traits
Macros
Functions
- autodiff::hvp
- autodiff::jvp
- autodiff::vmap
- autodiff::vmap::vmap
- available_devices
- benchmark_devices
- device_chain_from_env
- device_from_env
- device_label
- device_report
- devices_for
- devices_for_with_policy
- fastest_device
- fastest_device_for
- fusion_passes
- fusion_passes_for_supported
- graph_param_names
- hvp
- inspect_graph
- inspect_graph_diff
- inspect_hir
- inspect_hir_stats
- inspect_lir
- inspect_mir
- inspect_mir_diff
- inspect_mir_stats
- inspect_pipeline
- is_available
- jvp
- maybe_dump_pipeline
- node_label
- parse_device
- parse_device_list
- prelude::available_devices
- prelude::benchmark_devices
- prelude::device_chain_from_env
- prelude::device_from_env
- prelude::device_label
- prelude::device_report
- prelude::devices_for
- prelude::devices_for_with_policy
- prelude::env::clear_overrides
- prelude::env::flag
- prelude::env::is_unset
- prelude::env::normalize_key
- prelude::env::parse_or
- prelude::env::set
- prelude::env::unset
- prelude::env::var
- prelude::env::var_os
- prelude::fastest_device
- prelude::fastest_device_for
- prelude::flag
- prelude::graph_param_names
- prelude::hvp
- prelude::is_available
- prelude::jvp
- prelude::parse_device
- prelude::parse_device_list
- prelude::resolve_device
- prelude::resolve_device_chain
- prelude::run_with_fallback
- prelude::scalar_constant_bytes
- prelude::set
- prelude::unset
- prelude::var
- prelude::vmap
- prelude::vmap::vmap
- resolve_device
- resolve_device_chain
- run_with_fallback
- scalar_constant_bytes
- supported_for_target
- supports_op
- vmap
- vmap::vmap