List of all items
Structs
- graph::KernelExecutable
- graph::TritonLowering
- math::gemm::MatMulRuntimeOp
- math::gemm::MatmulBackwardDa
- math::gemm::MatmulBackwardDb
- math::gemm::MatmulForward
- nn::activation::elu::CeluBackward
- nn::activation::elu::CeluForward
- nn::activation::elu::CeluForwardDispatch
- nn::activation::elu::CeluOp
- nn::activation::elu::EluBackward
- nn::activation::elu::EluForward
- nn::activation::elu::EluForwardDispatch
- nn::activation::elu::EluOp
- nn::activation::elu::SeluBackward
- nn::activation::elu::SeluForward
- nn::activation::elu::SeluForwardDispatch
- nn::activation::elu::SeluOp
- nn::activation::extra::LogSoftmaxBackward
- nn::activation::extra::LogSoftmaxForward
- nn::activation::extra::PreluBackward
- nn::activation::extra::PreluForward
- nn::activation::extra::ShrinkBackward
- nn::activation::extra::ShrinkForward
- nn::activation::extra::ShrinkRuntimeOp
- nn::activation::extra::SwishBackward
- nn::activation::extra::SwishForward
- nn::activation::extra::ThresholdedReluBackward
- nn::activation::extra::ThresholdedReluForward
- nn::activation::extra::ThresholdedReluRuntimeOp
- nn::activation::gelu::GeluBackward
- nn::activation::gelu::GeluForward
- nn::activation::gelu::GeluForwardDispatch
- nn::activation::gelu::GeluOp
- nn::activation::gelu::MishBackward
- nn::activation::gelu::MishForward
- nn::activation::gelu::MishForwardDispatch
- nn::activation::gelu::MishOp
- nn::activation::hard::HardshrinkBackward
- nn::activation::hard::HardshrinkForward
- nn::activation::hard::HardshrinkForwardDispatch
- nn::activation::hard::HardshrinkOp
- nn::activation::hard::HardsigmoidBackward
- nn::activation::hard::HardsigmoidForward
- nn::activation::hard::HardsigmoidForwardDispatch
- nn::activation::hard::HardsigmoidOp
- nn::activation::hard::HardswishBackward
- nn::activation::hard::HardswishForward
- nn::activation::hard::HardswishForwardDispatch
- nn::activation::hard::HardswishOp
- nn::activation::hard::HardtanhBackward
- nn::activation::hard::HardtanhForward
- nn::activation::hard::HardtanhForwardDispatch
- nn::activation::hard::HardtanhOp
- nn::activation::hard::Relu6Backward
- nn::activation::hard::Relu6Forward
- nn::activation::hard::Relu6ForwardDispatch
- nn::activation::hard::Relu6Op
- nn::activation::misc::LeakyReluBackward
- nn::activation::misc::LeakyReluForward
- nn::activation::misc::LeakyReluForwardDispatch
- nn::activation::misc::LeakyReluOp
- nn::activation::misc::SoftplusBackward
- nn::activation::misc::SoftplusForward
- nn::activation::misc::SoftplusForwardDispatch
- nn::activation::misc::SoftplusOp
- nn::activation::misc::SoftshrinkBackward
- nn::activation::misc::SoftshrinkForward
- nn::activation::misc::SoftshrinkForwardDispatch
- nn::activation::misc::SoftshrinkOp
- nn::activation::misc::SoftsignBackward
- nn::activation::misc::SoftsignForward
- nn::activation::misc::SoftsignForwardDispatch
- nn::activation::misc::SoftsignOp
- nn::activation::misc::ThresholdBackward
- nn::activation::misc::ThresholdForward
- nn::activation::misc::ThresholdForwardDispatch
- nn::activation::misc::ThresholdOp
- nn::activation::relu::ReluBackward
- nn::activation::relu::ReluForward
- nn::activation::relu::ReluOp
- nn::activation::sigmoid::LogsigmoidBackward
- nn::activation::sigmoid::LogsigmoidForward
- nn::activation::sigmoid::LogsigmoidForwardDispatch
- nn::activation::sigmoid::LogsigmoidOp
- nn::activation::sigmoid::SigmoidBackward
- nn::activation::sigmoid::SigmoidForward
- nn::activation::sigmoid::SigmoidForwardDispatch
- nn::activation::sigmoid::SigmoidOp
- nn::activation::sigmoid::SiluBackward
- nn::activation::sigmoid::SiluForward
- nn::activation::sigmoid::SiluForwardDispatch
- nn::activation::sigmoid::SiluOp
- nn::activation::softmax::SoftmaxBackward
- nn::activation::softmax::SoftmaxForward
- nn::activation::softmax::SoftmaxOp
- nn::activation::tanh::TanhBackward
- nn::activation::tanh::TanhForward
- nn::activation::tanh::TanhForwardDispatch
- nn::activation::tanh::TanhOp
- nn::activation::tanh::TanhshrinkBackward
- nn::activation::tanh::TanhshrinkForward
- nn::activation::tanh::TanhshrinkForwardDispatch
- nn::activation::tanh::TanhshrinkOp
- nn::attention::flash_attn2::FlashAttention2BackwardDkv
- nn::attention::flash_attn2::FlashAttention2BackwardDq
- nn::attention::flash_attn2::FlashAttention2Forward
- nn::attention::flash_attn2::FlashAttention2Op
- nn::conv::conv1d::Conv1dBackwardDw
- nn::conv::conv1d::Conv1dBackwardDx
- nn::conv::conv1d::Conv1dForward
- nn::conv::conv1d::Conv1dOp
- nn::conv::conv2d::Conv2dBackward
- nn::conv::conv2d::Conv2dBackwardDw
- nn::conv::conv2d::Conv2dBackwardDx
- nn::conv::conv2d::Conv2dBiasForward
- nn::conv::conv2d::Conv2dForward
- nn::conv::conv2d::Conv2dOp
- nn::conv::conv3d::Conv3dBackwardDw
- nn::conv::conv3d::Conv3dBackwardDx
- nn::conv::conv3d::Conv3dForward
- nn::conv::conv3d::Conv3dOp
- nn::fused::conv2d_bn_silu::Conv2dBnSiluForward
- nn::fused::conv2d_bn_silu_gemm::Conv2dBnSiluGemmForward
- nn::fused::conv2d_bn_silu_tiled::Conv2dBnSiluTiledForward
- nn::loss::bce::BceLossBackward
- nn::loss::bce::BceLossForward
- nn::loss::bce::BceWithLogitsLossBackward
- nn::loss::bce::BceWithLogitsLossForward
- nn::loss::bce::GaussianNllLossBackwardInput
- nn::loss::bce::GaussianNllLossForward
- nn::loss::bce::KlDivLossBackward
- nn::loss::bce::KlDivLossForward
- nn::loss::bce::PoissonNllLossBackward
- nn::loss::bce::PoissonNllLossForward
- nn::loss::bce::SoftMarginLossBackward
- nn::loss::bce::SoftMarginLossForward
- nn::loss::elementwise::HuberLossBackward
- nn::loss::elementwise::HuberLossForward
- nn::loss::elementwise::L1LossBackward
- nn::loss::elementwise::L1LossForward
- nn::loss::elementwise::MseLossBackward
- nn::loss::elementwise::MseLossForward
- nn::loss::elementwise::SmoothL1LossBackward
- nn::loss::elementwise::SmoothL1LossForward
- nn::loss::embedding::CosineEmbeddingLossBackward
- nn::loss::embedding::CosineEmbeddingLossForward
- nn::loss::embedding::TripletMarginLossBackward
- nn::loss::embedding::TripletMarginLossForward
- nn::loss::nll::CrossEntropyLossBackward
- nn::loss::nll::CrossEntropyLossForward
- nn::loss::nll::MultilabelSoftMarginLossBackward
- nn::loss::nll::MultilabelSoftMarginLossForward
- nn::loss::nll::NllLossBackward
- nn::loss::nll::NllLossForward
- nn::loss::ranking::HingeEmbeddingLossBackward
- nn::loss::ranking::HingeEmbeddingLossForward
- nn::loss::ranking::MarginRankingLossBackward
- nn::loss::ranking::MarginRankingLossForward
- nn::loss::ranking::MultiMarginLossBackward
- nn::loss::ranking::MultiMarginLossForward
- nn::mlp::flatten::FlattenBackward
- nn::mlp::flatten::FlattenForward
- nn::mlp::flatten::FlattenOp
- nn::mlp::linear::LinearBackward
- nn::mlp::linear::LinearForward
- nn::mlp::linear::LinearOp
- nn::norm::batchnorm::BatchNorm2dNchwBackward
- nn::norm::batchnorm::BatchNorm2dNchwForwardInference
- nn::norm::batchnorm::BatchNorm2dNchwInferenceRuntimeOp
- nn::norm::batchnorm::BatchNormBackward
- nn::norm::batchnorm::BatchNormForwardInference
- nn::norm::batchnorm::BatchNormNormalizeForward
- nn::norm::batchnorm::BatchNormNormalizeRuntimeOp
- nn::norm::batchnorm::BatchNormStatsForward
- nn::norm::batchnorm::BatchNormStatsRuntimeOp
- nn::norm::groupnorm::GroupNormBackward
- nn::norm::groupnorm::GroupNormForward
- nn::norm::groupnorm::GroupNormForwardInference
- nn::norm::instancenorm::InstanceNormBackward
- nn::norm::instancenorm::InstanceNormForward
- nn::norm::instancenorm::InstanceNormForwardInference
- nn::norm::layernorm::LayerNormBackward
- nn::norm::layernorm::LayerNormForward
- nn::norm::layernorm::LayerNormForwardInference
- nn::norm::layernorm::LayerNormForwardInferenceRuntimeOp
- nn::norm::rmsnorm::RmsNormBackward
- nn::norm::rmsnorm::RmsNormForward
- nn::optim::adagrad::AdadeltaStep
- nn::optim::adagrad::AdagradStep
- nn::optim::adam::AdamStep
- nn::optim::adam::AdamwStep
- nn::optim::adamax::AdamaxStep
- nn::optim::asgd::AsgdStep
- nn::optim::muon::MuonFrobNormSq
- nn::optim::muon::MuonNsStep
- nn::optim::muon::MuonNsXtx
- nn::optim::muon::MuonUpdate
- nn::optim::nadam::NadamStep
- nn::optim::radam::RadamAdaptiveStep
- nn::optim::radam::RadamSgdStep
- nn::optim::rmsprop::RmspropMomentumStep
- nn::optim::rmsprop::RmspropStep
- nn::optim::rprop::RpropStep
- nn::optim::sgd::SgdMomentumStep
- nn::optim::sgd::SgdNesterovStep
- nn::optim::sgd::SgdStep
- nn::pad::circular_pad1d::CircularPad1dBackward
- nn::pad::circular_pad1d::CircularPad1dForward
- nn::pad::circular_pad1d::CircularPad1dOp
- nn::pad::circular_pad2d::CircularPad2dBackward
- nn::pad::circular_pad2d::CircularPad2dForward
- nn::pad::circular_pad2d::CircularPad2dOp
- nn::pad::circular_pad3d::CircularPad3dBackward
- nn::pad::circular_pad3d::CircularPad3dForward
- nn::pad::circular_pad3d::CircularPad3dOp
- nn::pad::constant_pad1d::ConstantPad1dBackward
- nn::pad::constant_pad1d::ConstantPad1dForward
- nn::pad::constant_pad1d::ConstantPad1dOp
- nn::pad::constant_pad2d::ConstantPad2dBackward
- nn::pad::constant_pad2d::ConstantPad2dForward
- nn::pad::constant_pad2d::ConstantPad2dOp
- nn::pad::constant_pad3d::ConstantPad3dBackward
- nn::pad::constant_pad3d::ConstantPad3dForward
- nn::pad::constant_pad3d::ConstantPad3dOp
- nn::pad::reflection_pad1d::ReflectionPad1dBackward
- nn::pad::reflection_pad1d::ReflectionPad1dForward
- nn::pad::reflection_pad1d::ReflectionPad1dOp
- nn::pad::reflection_pad2d::ReflectionPad2dBackward
- nn::pad::reflection_pad2d::ReflectionPad2dForward
- nn::pad::reflection_pad2d::ReflectionPad2dOp
- nn::pad::reflection_pad3d::ReflectionPad3dBackward
- nn::pad::reflection_pad3d::ReflectionPad3dForward
- nn::pad::reflection_pad3d::ReflectionPad3dOp
- nn::pad::replication_pad1d::ReplicationPad1dBackward
- nn::pad::replication_pad1d::ReplicationPad1dForward
- nn::pad::replication_pad1d::ReplicationPad1dOp
- nn::pad::replication_pad2d::ReplicationPad2dBackward
- nn::pad::replication_pad2d::ReplicationPad2dForward
- nn::pad::replication_pad2d::ReplicationPad2dOp
- nn::pad::replication_pad3d::ReplicationPad3dBackward
- nn::pad::replication_pad3d::ReplicationPad3dForward
- nn::pad::replication_pad3d::ReplicationPad3dOp
- nn::pool::avgpool1d::Avgpool1dBackward
- nn::pool::avgpool1d::Avgpool1dForward
- nn::pool::avgpool1d::Avgpool1dOp
- nn::pool::avgpool2d::Avgpool2dBackward
- nn::pool::avgpool2d::Avgpool2dForward
- nn::pool::avgpool2d::Avgpool2dOp
- nn::pool::avgpool3d::Avgpool3dBackward
- nn::pool::avgpool3d::Avgpool3dForward
- nn::pool::avgpool3d::Avgpool3dOp
- nn::pool::lppool1d::Lppool1dBackward
- nn::pool::lppool1d::Lppool1dForward
- nn::pool::lppool1d::Lppool1dOp
- nn::pool::lppool2d::Lppool2dBackward
- nn::pool::lppool2d::Lppool2dForward
- nn::pool::lppool2d::Lppool2dOp
- nn::pool::lppool3d::Lppool3dBackward
- nn::pool::lppool3d::Lppool3dForward
- nn::pool::lppool3d::Lppool3dOp
- nn::pool::maxpool1d::Maxpool1dBackward
- nn::pool::maxpool1d::Maxpool1dForward
- nn::pool::maxpool1d::Maxpool1dOp
- nn::pool::maxpool2d::Maxpool2dBackward
- nn::pool::maxpool2d::Maxpool2dForward
- nn::pool::maxpool2d::Maxpool2dOp
- nn::pool::maxpool3d::Maxpool3dBackward
- nn::pool::maxpool3d::Maxpool3dForward
- nn::pool::maxpool3d::Maxpool3dOp
- nn::tensor::channel_bias_add::ChannelBiasAddBackward
- nn::tensor::channel_bias_add::ChannelBiasAddForward
- nn::tensor::channel_bias_add::ChannelBiasAddRuntimeOp
- nn::tensor::channel_bias_add::NchwBiasAddBackward
- nn::tensor::channel_bias_add::NchwBiasAddForward
- nn::tensor::channel_bias_add::NchwBiasAddRuntimeOp
- nn::tensor::channel_cat::ChannelCatBackward
- nn::tensor::channel_cat::ChannelCatForward
- nn::tensor::channel_cat::ChannelCatOp
- nn::tensor::channel_cat::ChannelCatRuntimeOp
- nn::tensor::channel_chunk::ChannelChunkBackward
- nn::tensor::channel_chunk::ChannelChunkForward
- nn::tensor::channel_chunk::ChannelChunkOp
- nn::tensor::channel_chunk::ChannelChunkRuntimeOp
- nn::tensor::elemwise_add::ElemwiseAddBackward
- nn::tensor::elemwise_add::ElemwiseAddForward
- nn::tensor::elemwise_binary::ClipRuntimeOp
- nn::tensor::elemwise_binary::ElemwiseClipBackward
- nn::tensor::elemwise_binary::ElemwiseClipForward
- nn::tensor::elemwise_binary::ElemwiseDivBackward
- nn::tensor::elemwise_binary::ElemwiseDivForward
- nn::tensor::elemwise_binary::ElemwiseEqualForward
- nn::tensor::elemwise_binary::ElemwiseFmodForward
- nn::tensor::elemwise_binary::ElemwiseGreaterEqualForward
- nn::tensor::elemwise_binary::ElemwiseGreaterForward
- nn::tensor::elemwise_binary::ElemwiseLessEqualForward
- nn::tensor::elemwise_binary::ElemwiseLessForward
- nn::tensor::elemwise_binary::ElemwiseMaxBackward
- nn::tensor::elemwise_binary::ElemwiseMaxForward
- nn::tensor::elemwise_binary::ElemwiseMeanBackward
- nn::tensor::elemwise_binary::ElemwiseMeanForward
- nn::tensor::elemwise_binary::ElemwiseMinBackward
- nn::tensor::elemwise_binary::ElemwiseMinForward
- nn::tensor::elemwise_binary::ElemwiseMulBackward
- nn::tensor::elemwise_binary::ElemwiseMulForward
- nn::tensor::elemwise_binary::ElemwisePowBackward
- nn::tensor::elemwise_binary::ElemwisePowForward
- nn::tensor::elemwise_binary::ElemwiseSubBackward
- nn::tensor::elemwise_binary::ElemwiseSubForward
- nn::tensor::elemwise_binary::ElemwiseSumBackward
- nn::tensor::elemwise_binary::ElemwiseSumForward
- nn::tensor::elemwise_binary::ElemwiseWhereBackward
- nn::tensor::elemwise_binary::ElemwiseWhereForward
- nn::tensor::elemwise_unary::ElemwiseAbsBackward
- nn::tensor::elemwise_unary::ElemwiseAbsForward
- nn::tensor::elemwise_unary::ElemwiseAcosBackward
- nn::tensor::elemwise_unary::ElemwiseAcosForward
- nn::tensor::elemwise_unary::ElemwiseAcoshBackward
- nn::tensor::elemwise_unary::ElemwiseAcoshForward
- nn::tensor::elemwise_unary::ElemwiseAsinBackward
- nn::tensor::elemwise_unary::ElemwiseAsinForward
- nn::tensor::elemwise_unary::ElemwiseAsinhBackward
- nn::tensor::elemwise_unary::ElemwiseAsinhForward
- nn::tensor::elemwise_unary::ElemwiseAtanBackward
- nn::tensor::elemwise_unary::ElemwiseAtanForward
- nn::tensor::elemwise_unary::ElemwiseAtanhBackward
- nn::tensor::elemwise_unary::ElemwiseAtanhForward
- nn::tensor::elemwise_unary::ElemwiseCeilForward
- nn::tensor::elemwise_unary::ElemwiseCosBackward
- nn::tensor::elemwise_unary::ElemwiseCosForward
- nn::tensor::elemwise_unary::ElemwiseCoshBackward
- nn::tensor::elemwise_unary::ElemwiseCoshForward
- nn::tensor::elemwise_unary::ElemwiseErfBackward
- nn::tensor::elemwise_unary::ElemwiseErfForward
- nn::tensor::elemwise_unary::ElemwiseExpBackward
- nn::tensor::elemwise_unary::ElemwiseExpForward
- nn::tensor::elemwise_unary::ElemwiseFloorForward
- nn::tensor::elemwise_unary::ElemwiseIsnanForward
- nn::tensor::elemwise_unary::ElemwiseLogBackward
- nn::tensor::elemwise_unary::ElemwiseLogForward
- nn::tensor::elemwise_unary::ElemwiseNegBackward
- nn::tensor::elemwise_unary::ElemwiseNegForward
- nn::tensor::elemwise_unary::ElemwiseReciprocalBackward
- nn::tensor::elemwise_unary::ElemwiseReciprocalForward
- nn::tensor::elemwise_unary::ElemwiseSignForward
- nn::tensor::elemwise_unary::ElemwiseSinBackward
- nn::tensor::elemwise_unary::ElemwiseSinForward
- nn::tensor::elemwise_unary::ElemwiseSinhBackward
- nn::tensor::elemwise_unary::ElemwiseSinhForward
- nn::tensor::elemwise_unary::ElemwiseSqrtBackward
- nn::tensor::elemwise_unary::ElemwiseSqrtForward
- nn::tensor::elemwise_unary::ElemwiseTanBackward
- nn::tensor::elemwise_unary::ElemwiseTanForward
- nn::tensor::reduction::CumProdForward
- nn::tensor::reduction::CumSumForward
- nn::tensor::reduction::GlobalAvgPoolForward
- nn::tensor::reduction::GlobalMaxPoolForward
- nn::tensor::reduction::ReduceL1Forward
- nn::tensor::reduction::ReduceL2Forward
- nn::tensor::reduction::ReduceLogSumExpForward
- nn::tensor::reduction::ReduceLogSumForward
- nn::tensor::reduction::ReduceMaxForward
- nn::tensor::reduction::ReduceMeanForward
- nn::tensor::reduction::ReduceMinForward
- nn::tensor::reduction::ReduceProdForward
- nn::tensor::reduction::ReduceSumForward
- nn::tensor::reduction::ReduceSumSquareForward
- nn::tensor::upsample_nearest2d::UpsampleNearest2dBackward
- nn::tensor::upsample_nearest2d::UpsampleNearest2dForward
- nn::tensor::upsample_nearest2d::UpsampleNearest2dOp
Enums
Functions
- math::gemm::matmul_backward_da
- math::gemm::matmul_backward_db
- math::gemm::matmul_forward
- nn::activation::elu::celu_backward
- nn::activation::elu::celu_forward
- nn::activation::elu::elu_backward
- nn::activation::elu::elu_forward
- nn::activation::elu::selu_backward
- nn::activation::elu::selu_forward
- nn::activation::extra::log_softmax_backward
- nn::activation::extra::log_softmax_forward
- nn::activation::extra::prelu_backward
- nn::activation::extra::prelu_forward
- nn::activation::extra::shrink_backward
- nn::activation::extra::shrink_forward
- nn::activation::extra::swish_backward
- nn::activation::extra::swish_forward
- nn::activation::extra::thresholded_relu_backward
- nn::activation::extra::thresholded_relu_forward
- nn::activation::gelu::gelu_backward
- nn::activation::gelu::gelu_forward
- nn::activation::gelu::mish_backward
- nn::activation::gelu::mish_forward
- nn::activation::hard::hardshrink_backward
- nn::activation::hard::hardshrink_forward
- nn::activation::hard::hardsigmoid_backward
- nn::activation::hard::hardsigmoid_forward
- nn::activation::hard::hardswish_backward
- nn::activation::hard::hardswish_forward
- nn::activation::hard::hardtanh_backward
- nn::activation::hard::hardtanh_forward
- nn::activation::hard::relu6_backward
- nn::activation::hard::relu6_forward
- nn::activation::misc::leaky_relu_backward
- nn::activation::misc::leaky_relu_forward
- nn::activation::misc::softplus_backward
- nn::activation::misc::softplus_forward
- nn::activation::misc::softshrink_backward
- nn::activation::misc::softshrink_forward
- nn::activation::misc::softsign_backward
- nn::activation::misc::softsign_forward
- nn::activation::misc::threshold_backward
- nn::activation::misc::threshold_forward
- nn::activation::relu::relu_backward
- nn::activation::relu::relu_forward
- nn::activation::sigmoid::logsigmoid_backward
- nn::activation::sigmoid::logsigmoid_forward
- nn::activation::sigmoid::sigmoid_backward
- nn::activation::sigmoid::sigmoid_forward
- nn::activation::sigmoid::silu_backward
- nn::activation::sigmoid::silu_forward
- nn::activation::softmax::softmax_backward
- nn::activation::softmax::softmax_forward
- nn::activation::tanh::tanh_backward
- nn::activation::tanh::tanh_forward
- nn::activation::tanh::tanhshrink_backward
- nn::activation::tanh::tanhshrink_forward
- nn::attention::flash_attn2::flash_attention2_backward_dkv
- nn::attention::flash_attn2::flash_attention2_backward_dq
- nn::attention::flash_attn2::flash_attention2_forward
- nn::conv::conv1d::conv1d_backward_dw
- nn::conv::conv1d::conv1d_backward_dx
- nn::conv::conv1d::conv1d_forward
- nn::conv::conv2d::conv2d_backward
- nn::conv::conv2d::conv2d_backward_dw
- nn::conv::conv2d::conv2d_backward_dx
- nn::conv::conv2d::conv2d_bias_forward
- nn::conv::conv2d::conv2d_forward
- nn::conv::conv3d::conv3d_backward_dw
- nn::conv::conv3d::conv3d_backward_dx
- nn::conv::conv3d::conv3d_forward
- nn::fused::conv2d_bn_silu::conv2d_bn_silu_forward
- nn::fused::conv2d_bn_silu_gemm::conv2d_bn_silu_gemm_forward
- nn::fused::conv2d_bn_silu_tiled::conv2d_bn_silu_tiled_forward
- nn::loss::bce::bce_loss_backward
- nn::loss::bce::bce_loss_forward
- nn::loss::bce::bce_with_logits_loss_backward
- nn::loss::bce::bce_with_logits_loss_forward
- nn::loss::bce::gaussian_nll_loss_backward_input
- nn::loss::bce::gaussian_nll_loss_forward
- nn::loss::bce::kl_div_loss_backward
- nn::loss::bce::kl_div_loss_forward
- nn::loss::bce::poisson_nll_loss_backward
- nn::loss::bce::poisson_nll_loss_forward
- nn::loss::bce::soft_margin_loss_backward
- nn::loss::bce::soft_margin_loss_forward
- nn::loss::elementwise::huber_loss_backward
- nn::loss::elementwise::huber_loss_forward
- nn::loss::elementwise::l1_loss_backward
- nn::loss::elementwise::l1_loss_forward
- nn::loss::elementwise::mse_loss_backward
- nn::loss::elementwise::mse_loss_forward
- nn::loss::elementwise::smooth_l1_loss_backward
- nn::loss::elementwise::smooth_l1_loss_forward
- nn::loss::embedding::cosine_embedding_loss_backward
- nn::loss::embedding::cosine_embedding_loss_forward
- nn::loss::embedding::triplet_margin_loss_backward
- nn::loss::embedding::triplet_margin_loss_forward
- nn::loss::nll::cross_entropy_loss_backward
- nn::loss::nll::cross_entropy_loss_forward
- nn::loss::nll::multilabel_soft_margin_loss_backward
- nn::loss::nll::multilabel_soft_margin_loss_forward
- nn::loss::nll::nll_loss_backward
- nn::loss::nll::nll_loss_forward
- nn::loss::ranking::hinge_embedding_loss_backward
- nn::loss::ranking::hinge_embedding_loss_forward
- nn::loss::ranking::margin_ranking_loss_backward
- nn::loss::ranking::margin_ranking_loss_forward
- nn::loss::ranking::multi_margin_loss_backward
- nn::loss::ranking::multi_margin_loss_forward
- nn::mlp::flatten::flatten_backward
- nn::mlp::flatten::flatten_forward
- nn::mlp::linear::linear_backward
- nn::mlp::linear::linear_forward
- nn::norm::batchnorm::batch_norm_2d_nchw_backward
- nn::norm::batchnorm::batch_norm_2d_nchw_forward_inference
- nn::norm::batchnorm::batch_norm_backward
- nn::norm::batchnorm::batch_norm_forward_inference
- nn::norm::batchnorm::batch_norm_normalize_forward
- nn::norm::batchnorm::batch_norm_stats_forward
- nn::norm::groupnorm::group_norm_backward
- nn::norm::groupnorm::group_norm_forward
- nn::norm::groupnorm::group_norm_forward_inference
- nn::norm::instancenorm::instance_norm_backward
- nn::norm::instancenorm::instance_norm_forward
- nn::norm::instancenorm::instance_norm_forward_inference
- nn::norm::layernorm::layer_norm_backward
- nn::norm::layernorm::layer_norm_forward
- nn::norm::layernorm::layer_norm_forward_inference
- nn::norm::rmsnorm::rms_norm_backward
- nn::norm::rmsnorm::rms_norm_forward
- nn::optim::adagrad::adadelta_step
- nn::optim::adagrad::adagrad_step
- nn::optim::adam::adam_step
- nn::optim::adam::adamw_step
- nn::optim::adamax::adamax_step
- nn::optim::asgd::asgd_step
- nn::optim::muon::muon_frob_norm_sq
- nn::optim::muon::muon_ns_step
- nn::optim::muon::muon_ns_xtx
- nn::optim::muon::muon_update
- nn::optim::nadam::nadam_step
- nn::optim::radam::radam_adaptive_step
- nn::optim::radam::radam_sgd_step
- nn::optim::rmsprop::rmsprop_momentum_step
- nn::optim::rmsprop::rmsprop_step
- nn::optim::rprop::rprop_step
- nn::optim::sgd::sgd_momentum_step
- nn::optim::sgd::sgd_nesterov_step
- nn::optim::sgd::sgd_step
- nn::pad::circular_pad1d::circular_pad1d_backward
- nn::pad::circular_pad1d::circular_pad1d_forward
- nn::pad::circular_pad2d::circular_pad2d_backward
- nn::pad::circular_pad2d::circular_pad2d_forward
- nn::pad::circular_pad3d::circular_pad3d_backward
- nn::pad::circular_pad3d::circular_pad3d_forward
- nn::pad::constant_pad1d::constant_pad1d_backward
- nn::pad::constant_pad1d::constant_pad1d_forward
- nn::pad::constant_pad2d::constant_pad2d_backward
- nn::pad::constant_pad2d::constant_pad2d_forward
- nn::pad::constant_pad3d::constant_pad3d_backward
- nn::pad::constant_pad3d::constant_pad3d_forward
- nn::pad::reflection_pad1d::reflection_pad1d_backward
- nn::pad::reflection_pad1d::reflection_pad1d_forward
- nn::pad::reflection_pad2d::reflection_pad2d_backward
- nn::pad::reflection_pad2d::reflection_pad2d_forward
- nn::pad::reflection_pad3d::reflection_pad3d_backward
- nn::pad::reflection_pad3d::reflection_pad3d_forward
- nn::pad::replication_pad1d::replication_pad1d_backward
- nn::pad::replication_pad1d::replication_pad1d_forward
- nn::pad::replication_pad2d::replication_pad2d_backward
- nn::pad::replication_pad2d::replication_pad2d_forward
- nn::pad::replication_pad3d::replication_pad3d_backward
- nn::pad::replication_pad3d::replication_pad3d_forward
- nn::pool::avgpool1d::avgpool1d_backward
- nn::pool::avgpool1d::avgpool1d_forward
- nn::pool::avgpool2d::avgpool2d_backward
- nn::pool::avgpool2d::avgpool2d_forward
- nn::pool::avgpool3d::avgpool3d_backward
- nn::pool::avgpool3d::avgpool3d_forward
- nn::pool::lppool1d::lppool1d_backward
- nn::pool::lppool1d::lppool1d_forward
- nn::pool::lppool2d::lppool2d_backward
- nn::pool::lppool2d::lppool2d_forward
- nn::pool::lppool3d::lppool3d_backward
- nn::pool::lppool3d::lppool3d_forward
- nn::pool::maxpool1d::maxpool1d_backward
- nn::pool::maxpool1d::maxpool1d_forward
- nn::pool::maxpool2d::maxpool2d_backward
- nn::pool::maxpool2d::maxpool2d_forward
- nn::pool::maxpool3d::maxpool3d_backward
- nn::pool::maxpool3d::maxpool3d_forward
- nn::tensor::channel_bias_add::channel_bias_add_backward
- nn::tensor::channel_bias_add::channel_bias_add_forward
- nn::tensor::channel_bias_add::nchw_bias_add_backward
- nn::tensor::channel_bias_add::nchw_bias_add_forward
- nn::tensor::channel_cat::channel_cat_backward
- nn::tensor::channel_cat::channel_cat_forward
- nn::tensor::channel_chunk::channel_chunk_backward
- nn::tensor::channel_chunk::channel_chunk_forward
- nn::tensor::elemwise_add::elemwise_add_backward
- nn::tensor::elemwise_add::elemwise_add_forward
- nn::tensor::elemwise_binary::elemwise_clip_backward
- nn::tensor::elemwise_binary::elemwise_clip_forward
- nn::tensor::elemwise_binary::elemwise_div_backward
- nn::tensor::elemwise_binary::elemwise_div_forward
- nn::tensor::elemwise_binary::elemwise_equal_forward
- nn::tensor::elemwise_binary::elemwise_fmod_forward
- nn::tensor::elemwise_binary::elemwise_greater_equal_forward
- nn::tensor::elemwise_binary::elemwise_greater_forward
- nn::tensor::elemwise_binary::elemwise_less_equal_forward
- nn::tensor::elemwise_binary::elemwise_less_forward
- nn::tensor::elemwise_binary::elemwise_max_backward
- nn::tensor::elemwise_binary::elemwise_max_forward
- nn::tensor::elemwise_binary::elemwise_mean_backward
- nn::tensor::elemwise_binary::elemwise_mean_forward
- nn::tensor::elemwise_binary::elemwise_min_backward
- nn::tensor::elemwise_binary::elemwise_min_forward
- nn::tensor::elemwise_binary::elemwise_mul_backward
- nn::tensor::elemwise_binary::elemwise_mul_forward
- nn::tensor::elemwise_binary::elemwise_pow_backward
- nn::tensor::elemwise_binary::elemwise_pow_forward
- nn::tensor::elemwise_binary::elemwise_sub_backward
- nn::tensor::elemwise_binary::elemwise_sub_forward
- nn::tensor::elemwise_binary::elemwise_sum_backward
- nn::tensor::elemwise_binary::elemwise_sum_forward
- nn::tensor::elemwise_binary::elemwise_where_backward
- nn::tensor::elemwise_binary::elemwise_where_forward
- nn::tensor::elemwise_unary::elemwise_abs_backward
- nn::tensor::elemwise_unary::elemwise_abs_forward
- nn::tensor::elemwise_unary::elemwise_acos_backward
- nn::tensor::elemwise_unary::elemwise_acos_forward
- nn::tensor::elemwise_unary::elemwise_acosh_backward
- nn::tensor::elemwise_unary::elemwise_acosh_forward
- nn::tensor::elemwise_unary::elemwise_asin_backward
- nn::tensor::elemwise_unary::elemwise_asin_forward
- nn::tensor::elemwise_unary::elemwise_asinh_backward
- nn::tensor::elemwise_unary::elemwise_asinh_forward
- nn::tensor::elemwise_unary::elemwise_atan_backward
- nn::tensor::elemwise_unary::elemwise_atan_forward
- nn::tensor::elemwise_unary::elemwise_atanh_backward
- nn::tensor::elemwise_unary::elemwise_atanh_forward
- nn::tensor::elemwise_unary::elemwise_ceil_forward
- nn::tensor::elemwise_unary::elemwise_cos_backward
- nn::tensor::elemwise_unary::elemwise_cos_forward
- nn::tensor::elemwise_unary::elemwise_cosh_backward
- nn::tensor::elemwise_unary::elemwise_cosh_forward
- nn::tensor::elemwise_unary::elemwise_erf_backward
- nn::tensor::elemwise_unary::elemwise_erf_forward
- nn::tensor::elemwise_unary::elemwise_exp_backward
- nn::tensor::elemwise_unary::elemwise_exp_forward
- nn::tensor::elemwise_unary::elemwise_floor_forward
- nn::tensor::elemwise_unary::elemwise_isnan_forward
- nn::tensor::elemwise_unary::elemwise_log_backward
- nn::tensor::elemwise_unary::elemwise_log_forward
- nn::tensor::elemwise_unary::elemwise_neg_backward
- nn::tensor::elemwise_unary::elemwise_neg_forward
- nn::tensor::elemwise_unary::elemwise_reciprocal_backward
- nn::tensor::elemwise_unary::elemwise_reciprocal_forward
- nn::tensor::elemwise_unary::elemwise_sign_forward
- nn::tensor::elemwise_unary::elemwise_sin_backward
- nn::tensor::elemwise_unary::elemwise_sin_forward
- nn::tensor::elemwise_unary::elemwise_sinh_backward
- nn::tensor::elemwise_unary::elemwise_sinh_forward
- nn::tensor::elemwise_unary::elemwise_sqrt_backward
- nn::tensor::elemwise_unary::elemwise_sqrt_forward
- nn::tensor::elemwise_unary::elemwise_tan_backward
- nn::tensor::elemwise_unary::elemwise_tan_forward
- nn::tensor::reduction::cum_prod_forward
- nn::tensor::reduction::cum_sum_forward
- nn::tensor::reduction::global_avg_pool_forward
- nn::tensor::reduction::global_max_pool_forward
- nn::tensor::reduction::reduce_l1_forward
- nn::tensor::reduction::reduce_l2_forward
- nn::tensor::reduction::reduce_log_sum_exp_forward
- nn::tensor::reduction::reduce_log_sum_forward
- nn::tensor::reduction::reduce_max_forward
- nn::tensor::reduction::reduce_mean_forward
- nn::tensor::reduction::reduce_min_forward
- nn::tensor::reduction::reduce_prod_forward
- nn::tensor::reduction::reduce_sum_forward
- nn::tensor::reduction::reduce_sum_square_forward
- nn::tensor::upsample_nearest2d::upsample_nearest2d_backward
- nn::tensor::upsample_nearest2d::upsample_nearest2d_forward