List of all items
Structs
- BatchNorm2dParams
- CpuBackend
- GroupNormNhwcParams
- LayerNormLastDimParams
- ParallelElementwiseConfig
- ParallelMatmulConfig
- RmsNormLastDimParams
- SeparableConv2dParams
- ThreadedCpuBackend
- ThreadedCpuBackendConfig
Enums
Traits
Functions
- add
- add_reduce_dispatch
- add_with_config
- add_with_config_and_pool
- avg_pool2d_nhwc
- avg_pool2d_nhwc_with_config
- avg_pool2d_nhwc_with_config_and_pool
- batch_norm2d_nhwc
- batch_norm2d_nhwc_with_config
- batch_norm2d_nhwc_with_config_and_pool
- binary_same_shape_dispatch
- conv2d_nhwc
- conv2d_nhwc_with_config
- conv2d_nhwc_with_config_and_pool
- conv3d
- deformable_conv2d_nhwc
- depthwise_conv2d_nhwc
- depthwise_conv2d_nhwc_with_config
- depthwise_conv2d_nhwc_with_config_and_pool
- dropout
- embedding_lookup
- exp
- exp_slice_dispatch
- exp_with_config
- exp_with_config_and_pool
- flash_attention
- fma_slice_dispatch
- gelu
- group_norm_nhwc
- group_norm_nhwc_with_config
- group_norm_nhwc_with_config_and_pool
- layer_norm_last_dim
- layer_norm_last_dim_with_config
- layer_norm_last_dim_with_config_and_pool
- log_softmax_last_dim
- log_softmax_last_dim_with_config
- log_softmax_last_dim_with_config_and_pool
- logsumexp_last_dim
- logsumexp_last_dim_with_config
- logsumexp_last_dim_with_config_and_pool
- matmul_2d
- matmul_2d_sequential
- matmul_2d_with_config
- matmul_2d_with_config_and_pool
- matmul_2d_with_threads
- matmul_row_dispatch
- max_pool2d_nhwc
- max_pool2d_nhwc_with_config
- max_pool2d_nhwc_with_config_and_pool
- max_reduce_dispatch
- mish
- mul
- mul_with_config
- mul_with_config_and_pool
- relu
- relu_inplace
- relu_out
- relu_slice_dispatch
- relu_to_slice_dispatch
- relu_with_config
- relu_with_config_and_pool
- rms_norm_last_dim
- rms_norm_last_dim_with_config
- rms_norm_last_dim_with_config_and_pool
- scaled_dot_product_attention
- separable_conv2d_nhwc
- separable_conv2d_nhwc_with_config
- separable_conv2d_nhwc_with_config_and_pool
- sigmoid
- sigmoid_slice_dispatch
- sigmoid_with_config
- sigmoid_with_config_and_pool
- silu
- softmax_last_dim
- softmax_last_dim_with_config
- softmax_last_dim_with_config_and_pool
- sub
- sub_exp_slice_dispatch
- sub_with_config
- sub_with_config_and_pool
- tanh_act
- tanh_act_with_config
- tanh_act_with_config_and_pool
- tanh_slice_dispatch
- transpose_conv2d_nhwc