List of all items

Structs

• ExtensionShapeContext
• ad::context::ShapeGuard
• ad::context::ShapeGuardContext
• ad::context::ShapeGuardFailure
• ad::context::TensorMeta
• broadcast::BroadcastInputPlan
• config::DotGeneralConfig
• config::GatherConfig
• config::PadConfig
• config::ScatterConfig
• config::SliceConfig
• ext_op::ExtensionEffect
• sym_dim::SymDim
• sym_dim::SymDimConversionError

Enums

• ExtensionShapeError
• ShapeRelation
• ad::ADRuleError
• ad::ADRuleKind
• ad::PrimitiveTransposeInput
• ad::context::MetadataRegistryError
• ad::context::ShapeGuardError
• axis::AxisError
• broadcast::BroadcastError
• config::CompareDir
• dim_expr::DimExpr
• dim_expr::DimExprEvalError
• ext_op::ExtensionAlias
• ext_op::ExtensionAliasDeclaration
• ext_op::ExtensionEffectAccess
• ext_op::ExtensionEffectDeclaration
• ext_op::ExtensionLoweringError
• ext_op::ExtensionStandardLowering
• input_key::TensorInputKey
• reduction::ReductionShapeError
• shape_extent::ShapeExtent
• std_tensor_op::StdTensorOp

Traits

• ad::PrimitiveRuleBuilder
• ext_op::ExtensionOp
• std_tensor_op::ConstantScalar

Derive Macros

• ExtensionFamilyId

Functions

• ad::context::lookup_global_metadata
• ad::linearize
• ad::transpose_rule
• axis::normalize_axes
• axis::normalize_axis
• broadcast::broadcast_error_to_validation
• broadcast::broadcast_in_dim_extent_error
• broadcast::broadcast_input_plan
• broadcast::broadcast_shape
• broadcast::broadcast_shapes
• reduction::reduced_shape

Type Aliases

• ad::ADRuleResult
• ad::context::ShapeGuardResult
• ext_op::ExtensionLoweringResult