List of all items

Structs

• quantization::HostQTensor

Enums

• binary::BinaryOp
• simd::CmpOp

Functions

• binary::binary_op
• binary::binary_op_typed
• binary::dispatch_float::float_add
• binary::dispatch_float::float_add_scalar
• binary::dispatch_float::float_atan2
• binary::dispatch_float::float_div
• binary::dispatch_float::float_div_scalar
• binary::dispatch_float::float_mul
• binary::dispatch_float::float_mul_scalar
• binary::dispatch_float::float_powf
• binary::dispatch_float::float_powf_scalar
• binary::dispatch_float::float_powf_scalar_impl
• binary::dispatch_float::float_powi
• binary::dispatch_float::float_powi_scalar
• binary::dispatch_float::float_remainder
• binary::dispatch_float::float_remainder_scalar
• binary::dispatch_float::float_sub
• binary::dispatch_float::float_sub_scalar
• binary::dispatch_int::bitwise_and
• binary::dispatch_int::bitwise_and_scalar
• binary::dispatch_int::bitwise_left_shift
• binary::dispatch_int::bitwise_left_shift_scalar
• binary::dispatch_int::bitwise_not
• binary::dispatch_int::bitwise_or
• binary::dispatch_int::bitwise_or_scalar
• binary::dispatch_int::bitwise_right_shift
• binary::dispatch_int::bitwise_right_shift_scalar
• binary::dispatch_int::bitwise_xor
• binary::dispatch_int::bitwise_xor_scalar
• binary::dispatch_int::int_add
• binary::dispatch_int::int_add_scalar
• binary::dispatch_int::int_div
• binary::dispatch_int::int_div_scalar
• binary::dispatch_int::int_mul
• binary::dispatch_int::int_mul_scalar
• binary::dispatch_int::int_powi
• binary::dispatch_int::int_powi_scalar
• binary::dispatch_int::int_powi_scalar_impl
• binary::dispatch_int::int_remainder
• binary::dispatch_int::int_remainder_scalar
• binary::dispatch_int::int_sub
• binary::dispatch_int::int_sub_scalar
• binary::int_binary_op
• binary::int_scalar_op
• binary::scalar_op
• binary::scalar_op_typed
• boolean::bool_and
• boolean::bool_argwhere
• boolean::bool_equal
• boolean::bool_equal_elem
• boolean::bool_not
• boolean::bool_ones
• boolean::bool_or
• boolean::bool_select_or
• boolean::bool_xor
• cast::bool_into_float
• cast::bool_into_int
• cast::float_cast
• cast::float_into_int
• cast::int_cast
• cast::int_into_float
• cat::cat
• comparison::all_bool
• comparison::all_bool_dim
• comparison::all_float
• comparison::all_float_dim
• comparison::all_int
• comparison::all_int_dim
• comparison::any_bool
• comparison::any_bool_dim
• comparison::any_float
• comparison::any_float_dim
• comparison::any_int
• comparison::any_int_dim
• comparison::bool_not_equal
• comparison::bool_not_equal_elem
• comparison::compare
• comparison::compare_elem
• comparison::dispatch::float_equal_elem
• comparison::dispatch::float_greater_elem
• comparison::dispatch::float_greater_equal_elem
• comparison::dispatch::float_lower_elem
• comparison::dispatch::float_lower_equal_elem
• comparison::dispatch::float_not_equal_elem
• comparison::dispatch::int_equal_elem
• comparison::dispatch::int_greater_elem
• comparison::dispatch::int_greater_equal_elem
• comparison::dispatch::int_lower_elem
• comparison::dispatch::int_lower_equal_elem
• comparison::dispatch::int_not_equal_elem
• comparison::equal
• comparison::equal_elem
• comparison::greater
• comparison::greater_elem
• comparison::greater_equal
• comparison::greater_equal_elem
• comparison::int_equal
• comparison::int_equal_elem
• comparison::int_greater
• comparison::int_greater_elem
• comparison::int_greater_equal
• comparison::int_greater_equal_elem
• comparison::int_lower
• comparison::int_lower_elem
• comparison::int_lower_equal
• comparison::int_lower_equal_elem
• comparison::int_not_equal
• comparison::int_not_equal_elem
• comparison::lower
• comparison::lower_elem
• comparison::lower_equal
• comparison::lower_equal_elem
• comparison::make_bool_tensor
• comparison::not_equal
• comparison::not_equal_elem
• cumulative::cummax
• cumulative::cummax_f32
• cumulative::cummax_f64
• cumulative::cummax_half
• cumulative::cummin
• cumulative::cummin_f32
• cumulative::cummin_f64
• cumulative::cummin_half
• cumulative::cumprod
• cumulative::cumprod_f32
• cumulative::cumprod_f64
• cumulative::cumprod_half
• cumulative::cumsum
• cumulative::cumsum_f32
• cumulative::cumsum_f64
• cumulative::cumsum_half
• cumulative::dispatch::float_cummax
• cumulative::dispatch::float_cummin
• cumulative::dispatch::float_cumprod
• cumulative::dispatch::float_cumsum
• cumulative::dispatch::int_cummax
• cumulative::dispatch::int_cummin
• cumulative::dispatch::int_cumprod
• cumulative::dispatch::int_cumsum
• expand::broadcast_binary
• expand::broadcast_shape
• expand::expand
• fill::float_full
• fill::float_ones
• fill::int_arange_step
• fill::int_full
• fill::int_ones
• flip::flip
• gather_scatter::dispatch_float::float_gather
• gather_scatter::dispatch_float::float_gather_nd
• gather_scatter::dispatch_float::float_scatter_add
• gather_scatter::dispatch_float::float_scatter_nd
• gather_scatter::dispatch_float::float_select
• gather_scatter::dispatch_float::float_select_add
• gather_scatter::dispatch_int::int_gather
• gather_scatter::dispatch_int::int_gather_nd
• gather_scatter::dispatch_int::int_scatter_add
• gather_scatter::dispatch_int::int_scatter_nd
• gather_scatter::dispatch_int::int_select
• gather_scatter::dispatch_int::int_select_add
• gather_scatter::gather
• gather_scatter::gather_bool
• gather_scatter::gather_f32
• gather_scatter::gather_f64
• gather_scatter::gather_i64
• gather_scatter::gather_nd
• gather_scatter::scatter_add
• gather_scatter::scatter_add_f32
• gather_scatter::scatter_add_f64
• gather_scatter::scatter_add_i64
• gather_scatter::scatter_nd
• gather_scatter::scatter_or
• gather_scatter::select
• gather_scatter::select_add
• gather_scatter::select_add_f32
• gather_scatter::select_add_f64
• gather_scatter::select_add_i64
• gather_scatter::select_f32
• gather_scatter::select_f64
• gather_scatter::select_i64
• mask::dispatch::float_mask_fill
• mask::dispatch::float_mask_where
• mask::dispatch::int_mask_fill
• mask::dispatch::int_mask_where
• mask::mask_fill
• mask::mask_fill_bf16
• mask::mask_fill_bool
• mask::mask_fill_f16
• mask::mask_fill_f32
• mask::mask_fill_f64
• mask::mask_fill_i64
• mask::mask_fill_u64
• mask::mask_where
• mask::mask_where_bf16
• mask::mask_where_bool
• mask::mask_where_f16
• mask::mask_where_f32
• mask::mask_where_f64
• mask::mask_where_i64
• quantization::dequantize
• quantization::q_argmax
• quantization::q_argmin
• quantization::q_expand
• quantization::q_flip
• quantization::q_from_data
• quantization::q_gather
• quantization::q_into_data
• quantization::q_permute
• quantization::q_reshape
• quantization::q_select
• quantization::q_slice
• quantization::q_swap_dims
• quantization::quantize
• quantization::quantize_dynamic
• reduce::argmax
• reduce::argmin
• reduce::dispatch::float_argmax
• reduce::dispatch::float_argmin
• reduce::dispatch::float_max_abs
• reduce::dispatch::float_max_abs_dim
• reduce::dispatch::float_max_dim_with_indices
• reduce::dispatch::float_min_dim_with_indices
• reduce::dispatch::int_max_abs
• reduce::dispatch::int_max_abs_dim
• reduce::dispatch::int_mean
• reduce::max
• reduce::max_dim
• reduce::max_dim_with_indices
• reduce::mean
• reduce::mean_dim
• reduce::min
• reduce::min_dim
• reduce::min_dim_with_indices
• reduce::prod
• reduce::prod_dim
• reduce::sum
• reduce::sum_dim
• repeat_dim::repeat_dim
• simd::abs_inplace_f32
• simd::add_inplace_f32
• simd::add_shared_row_inplace_f32
• simd::aligned::alloc_aligned
• simd::aligned::alloc_aligned_zeroed
• simd::aligned::from_slice
• simd::aligned::to_vec
• simd::bool_and_inplace_u8
• simd::bool_and_u8
• simd::bool_not_inplace_u8
• simd::bool_not_u8
• simd::bool_or_inplace_u8
• simd::bool_or_u8
• simd::bool_xor_inplace_u8
• simd::bool_xor_u8
• simd::cmp_f32
• simd::cmp_scalar_f32
• simd::div_inplace_f32
• simd::div_shared_row_inplace_f32
• simd::kernels::max_f32
• simd::kernels::min_f32
• simd::kernels::scatter_add_batched
• simd::kernels::scatter_add_f32
• simd::kernels::sum_f32
• simd::kernels::sum_rows_f32
• simd::mask_fill_f32
• simd::mask_fill_f64
• simd::mask_fill_i64
• simd::mask_fill_u8
• simd::mask_where_f32
• simd::mask_where_f64
• simd::mask_where_i64
• simd::mask_where_u8
• simd::mul_inplace_f32
• simd::mul_shared_row_inplace_f32
• simd::recip_inplace_f32
• simd::sub_inplace_f32
• simd::sub_shared_row_inplace_f32
• slice::slice
• slice::slice_assign
• sort::argsort
• sort::argtopk
• sort::dispatch::float_argsort
• sort::dispatch::float_argtopk
• sort::dispatch::float_sort_with_indices
• sort::dispatch::int_argtopk
• sort::sort
• sort::sort_with_indices
• unary::abs
• unary::acos
• unary::acosh
• unary::asin
• unary::asinh
• unary::atan
• unary::atanh
• unary::ceil
• unary::cos
• unary::cosh
• unary::dispatch_float::float_clamp
• unary::dispatch_float::float_clamp_max
• unary::dispatch_float::float_clamp_min
• unary::dispatch_float::float_is_inf
• unary::dispatch_float::float_is_nan
• unary::dispatch_float::float_neg
• unary::dispatch_float::float_sign
• unary::dispatch_int::int_clamp
• unary::dispatch_int::int_clamp_max
• unary::dispatch_int::int_clamp_min
• unary::dispatch_int::int_neg
• unary::dispatch_int::int_sign
• unary::erf
• unary::erf_f32
• unary::erf_f64
• unary::exp
• unary::float_predicate
• unary::floor
• unary::int_abs
• unary::log
• unary::log1p
• unary::recip
• unary::round
• unary::sin
• unary::sinh
• unary::sqrt
• unary::tan
• unary::tanh
• unary::trunc
• unary::unary_op
• unfold::unfold
• unfold::unfold_bool
• unfold::unfold_f32
• unfold::unfold_f64
• unfold::unfold_int

Type Aliases

• simd::aligned::AlignedVec

Constants

• simd::aligned::SIMD_ALIGN