Skip to content

Tracking issue for shortcuts to MLIR ops #273

@mofeing

Description

@mofeing
  • stablehlo

    • abs
    • add
    • after_all
    • all_gather
    • all_reduce
    • all_to_all
    • and
    • atan2
    • batch_norm_grad
    • batch_norm_inference
    • batch_norm_training
    • bitcast_convert
    • broadcast_in_dim
    • broadcast -> on deprecation process
    • case
    • cbrt
    • ceil
    • cholesky
    • clamp
    • count_leading_zeros
    • collective_broadcast
    • collective_permute
    • compare
    • complex
    • composite
    • concatenate
    • constant
    • convert refactor: use Ops instead of direct stablehlo calls #347
    • convolution add Ops.convolution #428
    • cosine
    • create_token -> on deprecation process
    • cross_replica_sum -> on deprecation process
    • custom_call
    • divide
    • dot_general add shortcut to dot_general op #330
    • dot -> on deprecation process
    • dynamic_broadcast_in_dim
    • dynamic_conv
    • dynamic_gather
    • dynamic_iota
    • dynamic_pad
    • dynamic_reshape
    • dynamic_slice
    • dynamic_update_slice
    • einsum -> on deprecation process, might be moved to CHLO
    • exponential
    • exponential_minus_one
    • fft
    • floor
    • gather
    • get_dimension_size
    • get_tuple_element -> on deprecation process
    • if_
    • imag
    • infeed
    • iota
    • is_finite
    • log_plus_one
    • log
    • logistic
    • map -> on deprecation process
    • maximum
    • minimum
    • multiply
    • negate
    • not
    • optimization_barrier
    • or
    • outfeed
    • pad
    • partition_id
    • popcnt
    • power
    • real_dynamic_slice -> on deprecation process
    • real
    • recv
    • reduce
    • reduce_precision
    • reduce_scatter
    • reduce_window
    • remainder
    • replica_id
    • reshape
    • reverse
    • rng_bit_generator
    • rng -> on deprecation process
    • round_nearest_even
    • round_nearest_afz
    • rsqrt
    • scatter
    • select_and_scatter
    • select add select op shortcut #328
    • send
    • set_dimension_size
    • shift_left
    • shift_right_arithmetic
    • shift_right_logical
    • sign
    • sine
    • slice
    • sort
    • sqrt
    • subtract
    • tan
    • tanh
    • torch_index_select -> on deprecation process
    • transpose
    • triangular_solve
    • tuple -> on deprecation process
    • unary_einsum -> on deprecation process, but used by Tenet
    • uniform_dequantize
    • uniform_quantize
    • while_
    • xor
  • chlo

    • acos
    • acosh
    • _asin_acos_kernel --> should not be used directly
    • asin
    • asinh
    • atan
    • atanh
    • bessel_i1e
    • broadcast_add
    • broadcast_and
    • broadcast_atan2
    • broadcast_compare
    • broadcast_complex
    • broadcast_divide
    • broadcast_maximum
    • broadcast_minimum
    • broadcast_multiply
    • broadcast_next_after
    • broadcast_or
    • broadcast_polygamma
    • broadcast_power
    • broadcast_remainder
    • broadcast_select
    • broadcast_shift_left
    • broadcast_shift_right_arithmetic
    • broadcast_shift_right_logical
    • broadcast_subtract
    • broadcast_xor
    • broadcast_zeta
    • conj
    • constant_like
    • constant
    • cosh
    • digamma
    • erf_inv
    • erf
    • erfc
    • is_inf
    • is_neg_inf
    • is_pos_inf
    • lgamma
    • next_after
    • polygamma
    • sinh
    • tan
    • top_k
    • zeta
  • enzyme

    • addTo
    • autodiff
    • batch
    • fwddiff
    • genericAdjoint
    • get
    • init
    • placeholder
    • pop
    • push
    • set

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions