List of all items
Structs
- KernelInfo
- Tensor
- TensorArangeWithDtypeBuilder
- TensorCumprodWithBuilder
- TensorCumsumWithBuilder
- activation::TensorBatchnormBuilder
- conditional::TensorClampBuilder
- conditional::TensorClipBuilder
- config::PrepareConfig
- data::TensorFromSliceWithBuilder
- error::AxisOutOfRangeSnafu
- error::BatchOutputMismatchSnafu
- error::BroadcastFewerDimensionsSnafu
- error::BroadcastIncompatibleSnafu
- error::BufferNotFoundSnafu
- error::BufferPtrNoSizeSnafu
- error::CompileKernelSnafu
- error::ConflictingReductionOptionsSnafu
- error::CreateProgramSnafu
- error::DependencyCyclesSnafu
- error::DeviceFactorySnafu
- error::DeviceSnafu
- error::DivisibilitySnafu
- error::DotDimensionSnafu
- error::DotShapeMismatchSnafu
- error::EmptyScheduleSnafu
- error::ExecutionSnafu
- error::ExpandDimensionMismatchSnafu
- error::ExpectedCallableOpSnafu
- error::ExpectedPtrDtypeSnafu
- error::FloatDTypeRequiredSnafu
- error::InvalidPermutationSnafu
- error::IrConstructionSnafu
- error::MultipleInferDimensionsSnafu
- error::NdarrayShapeSnafu
- error::NdimExactSnafu
- error::NdimMinimumSnafu
- error::NegativeDimensionSnafu
- error::NoBufferSnafu
- error::NoKernelsFoundSnafu
- error::NoShapeSnafu
- error::OptimizeSnafu
- error::ParamRangeSnafu
- error::PermutationLengthMismatchSnafu
- error::RangeifySnafu
- error::RenderKernelSnafu
- error::ReshapeSizeMismatchSnafu
- error::ShapeMismatchSnafu
- error::ShapeUnknownSnafu
- error::SqueezeDimensionNotOneSnafu
- error::SymbolicShapeSnafu
- error::SymbolicShapeUnsupportedSnafu
- error::TypeMismatchSnafu
- error::UOpSnafu
- error::VariableOutOfRangeSnafu
- matmul::TensorGemmBuilder
- matmul::TensorLinearBuilder
- matmul::TensorMatmulWithBuilder
- memory_planner::BufferKey
- memory_planner::BufferLiveness
- memory_planner::BufferPoolKey
- memory_planner::MemoryPlannerResult
- memory_planner::ReuseDependency
- nn::Conv1d
- nn::GruOutput
- nn::LSTMCell
- nn::Linear
- nn::LstmOutput
- nn::Relu
- nn::RnnOutput
- nn::TensorAvgPoolBuilder
- nn::TensorConvBuilder
- nn::TensorConvTransposeBuilder
- nn::TensorDepthToSpaceBuilder
- nn::TensorDropoutBuilder
- nn::TensorLpPoolBuilder
- nn::TensorLrnBuilder
- nn::TensorMaxPoolBuilder
- nn::TensorNllLossBuilder
- nn::pad::TensorPadWithBuilder
- reduce::TensorAllWithBuilder
- reduce::TensorAnyWithBuilder
- reduce::TensorArgmaxWithBuilder
- reduce::TensorArgminWithBuilder
- reduce::TensorMaxWithBuilder
- reduce::TensorMeanWithBuilder
- reduce::TensorMinWithBuilder
- reduce::TensorProdWithBuilder
- reduce::TensorStdMeanWithBuilder
- reduce::TensorStdWithBuilder
- reduce::TensorSumWithBuilder
- reduce::TensorVarMeanWithBuilder
- reduce::TensorVarWithBuilder
- schedule::BoundRangeRef
- schedule::KernelInvocation
- schedule::PreSchedule
- schedule::PreScheduleItem
- schedule::ScheduleItem
- schedule::ScheduleResult
- shape_ops::TensorSliceWithBuilder
- tensor_registry::TensorEntry
- transformer::TensorScaledDotProductAttentionBuilder
- variable::BoundVariable
- variable::Variable
Enums
- CpuBackend
- error::Error
- indexing::ScatterReduction
- memory_planner::PlannerMode
- nn::AspectRatioPolicy
- nn::AutoPad
- nn::CoordinateTransformMode
- nn::DepthToSpaceMode
- nn::GridSampleMode
- nn::GridSamplePaddingMode
- nn::NearestMode
- nn::PadMode
- nn::Reduction
- nn::ResizeMode
- reduce::AxisSpec
- shape_ops::MeshgridIndexing
Traits
Macros
Functions
- memory_planner::apply_buffer_replacements
- memory_planner::apply_reuse_dependencies
- memory_planner::memory_planner
- memory_planner::mode_from_env
- memory_planner::parse_mode
- nn::pad::auto_pad_split
- nn::pad::flat_pads_to_pairs
- nn::pad::resolve_pool_pads
- rand::manual_seed
- schedule::create_pre_schedule
- schedule::create_schedule
- schedule::instantiate_schedule
- tensor_registry::apply_map_to_tensors
- tensor_registry::apply_map_to_tensors_walk
- tensor_registry::buffer_count
- tensor_registry::gc_dead_refs
- tensor_registry::gc_unused_tensors
- tensor_registry::get_buffer
- tensor_registry::get_buffer_arc
- tensor_registry::get_tensor
- tensor_registry::register_buffer
- tensor_registry::register_buffer_by_uop_id
- tensor_registry::register_tensor
- tensor_registry::register_tensor_with_buffer
- tensor_registry::remove_buffer