Open Aarsh2001 opened 2 years ago
Add Operator functions for Jax frontend:
_
Please keep in mind that the proper way to link an issue to this list is to comment "- [ ] #issue_number" while the issue's title only includes the name of the function you've chosen. _
approx_max_k
approx_min_k
bessel_i0e
bessel_ile
betainc
bitcast_convert_type
population_count
broadcasted_iota
broadcast_in_dim
conv_dimension_numbers
conv_general_dilated
conv_general_dilated_local
conv_general_dilated_patches
conv_with_general_padding
dot_general
dynamic_iindex_in_dim
dynamic_slice
dynamic_slice_in_dim
dynamic_update_slice
dynamic_update_index_in_dim
dynamic_update_slice_in_dim
erf_inv #17021
fft
full
gather
ge
igammac
index_in_dim
index_take
https://github.com/unifyai/ivy/issues/14400
reduce
reduce_precision
reduce_window
rem
scatter
scatter_add
scatter_max
scatter_min
scatter_mul
select
shift_right_arithmetic
shift_right_logical
https://github.com/unifyai/ivy/issues/13767
sinh
sort
sort_key_val
square
sub
The main file paths where these functions are likely to be added are:
ivy/functional/frontends/jax/lax/operators.py ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_jax\_lax\_operators.py ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_lax/test\_operators.py
ivy/functional/frontends/jax/lax/operators.py
ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_jax\_lax\_operators.py
ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_lax/test\_operators.py
Sort is implemented in this commit: https://github.com/unifyai/ivy/commit/6d1968be2e4433f915baf556e5cda3caf97a33a0
Add Operator functions for Jax frontend:
_
Please keep in mind that the proper way to link an issue to this list is to comment "- [ ] #issue_number" while the issue's title only includes the name of the function you've chosen. _
2568
3297
2773
approx_max_k
approx_min_k
3106
3107
3088
7258
3090
3091
7260
17065
bessel_i0e
bessel_ile
betainc
bitcast_convert_type
3318
3109
3110
3111
population_count
3315
broadcasted_iota
broadcast_in_dim
20875
3115
3490
26732
21788
3491
14487
4253
3309
conv_dimension_numbers
conv_general_dilated
conv_general_dilated_local
conv_general_dilated_patches
conv_with_general_padding
4271
3481
3112
4575
21348
3312
3313
26379
3764
3763
dot_general
dynamic_iindex_in_dim
dynamic_slice
dynamic_slice_in_dim
dynamic_update_slice
dynamic_update_index_in_dim
dynamic_update_slice_in_dim
3117
12346
20988
erf_inv #17021
3307
4956
3765
fft
full
3176
gather
ge
3629
26750
igammac
20827
index_in_dim
index_take
21326
13916
3493
3541
26798
3494
3560
2456
3108
3119
3507
3174
14489
https://github.com/unifyai/ivy/issues/14400
3540
12391
2869
reduce
reduce_precision
reduce_window
3184
rem
3630
3113
3767
scatter
26894
scatter_add
scatter_max
scatter_min
scatter_mul
select
4796
shift_right_arithmetic
shift_right_logical
13402
https://github.com/unifyai/ivy/issues/13767
3092
2886
sinh
sort
sort_key_val
2641
square
13174
sub
3692
20991
11885
3766
The main file paths where these functions are likely to be added are:
ivy/functional/frontends/jax/lax/operators.py
ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_jax\_lax\_operators.py
ivy\_tests/test\_ivy/test\_frontends/test\_jax/test\_lax/test\_operators.py