Add missing frontend functions to compile popular models #9561
Labels
JAX Frontend
Developing the JAX Frontend, checklist triggered by commenting add_frontend_checklist
PyTorch Frontend
Developing the PyTorch Frontend, checklist triggered by commenting add_frontend_checklist
TensorFlow Frontend
Developing the TensorFlow Frontend, checklist triggered by commenting add_frontend_checklist
ToDo
A ToDo list of tasks
This list holds some frontend functions used in popular models that are currently missing. When adding new function to this ToDo list do so in alphabetical order (within each framework) 🙂
Torch
torch.nn.functional.interpolate #10523
torch.nn.functional.max_pool2d #10378
torch.nn.functional.conv_transpose3d #10357
torch.where #11669
torch.Tensor.__matmul__ #15048
torch.Tensor.__pow__ #11966
torch.Tensor.reshape_as #10148
torch.sigmoid #10599
torch.Tensor.sigmoid #10151
torch.split #10138
torch.Tensor.split #10137
torch.Tensor.expand_as #10324
torch.Tensor.repeat #11184
torch.histc #14474
torch.Tensor.median #14471
torch.functional.unique_consecutive #15288
torch.functional.norm
torch.norm #14816
torch.Tensor.norm #15268
torch.nn.functional.adaptive_max_pool2d #21224
torch.Tensor.reciprocal #14472
torch.nn.functional.grid_sample
torch.nn.functional.affine_grid
torch.Tensor.__and__ #12928
torch.Tensor.fill_ #14628
torch.linalg.lu_factor_ex
torch.linalg.lu_solve
torch.Tensor.masked_scatter
torch.Tensor.masked_scatter_
torch.masked_select
torch.Tensor.exp #14459
torch.functional.F.avg_pool3d #14431
torch.nn.functional.adaptive_max_pool3d
torch.Tensor.mul #14458
torch.select
torch.Tensor.remainder #14496
torch.Tensor.mm #14630
torch.Tensor.t #13553
torch.diag_embed
torch.Tensor.relu #13374
torch.cdist
torch.Tensor.cross #14460
torch.Tensor.det #14463
torch.bucketize
torch.Tensor.repeat_interleave
torch.Tensor.__rpow__ #14473
torch.Tensor.topk #15287
torch.Tensor.nonzero #14629
torch.Tensor.T #14855
torch.Tensor.floor #13555
torch.Tensor.round #14457
torch.bernoulli #14865
torch.Tensor.bernoulli_
torch.Tensor.index_put #21850
torch.randperm #14867
torch.nn.functional.local_response_norm
torch.nn.functional.instance_norm
torch.Tensor.prod #14706
torch.randn_like #14863
torch.Size.__add__ #15279
torch.Tensor.normal_ #14631
torch.Tensor.half #15265
torch.Tensor.mul_ #14469
torch.Tensor.div_ #14707
torch.Tensor.addcdiv #14864
torch.Tensor.sigmoid_ #14470
torch.Tensor.fmod_ #14946
TF
tensorflow.compat.v1.nn.fused_batch_norm
tensorflow.compat.v1.nn.separable_conv2d #11218
tensorflow.compat.v1.pad
tensorflow.nn.relu #9711
tensorflow.nn.relu6 #10567
tensorflow.compat.v1.pad #10215
tensorflow.floor #9706
tensorflow.image.extract_patches #10224
tensorflow.math.log #11790
tensorflow.image.resize #11804
tensorflow.math.ceil #9707
tensorflow.matmul #10075
tensorflow.minimum #9723
tensorflow.nn.softmax #9713
tensorflow.random.uniform #9726
tensorflow.realdiv #9715
tensorflow.roll #10610
tensorflow.sigmoid #9725
tensorflow.slice #10213
tf.Variable.__mul__
JAX
jax.numpy.take_along_axis #11792
jax.numpy.where #14464
The text was updated successfully, but these errors were encountered: