search

justinchuby / torch-onnx

Prototype of the next torch exporter
MIT License
1 stars 1 forks source link

Missing ops #53

Closed justinchuby closed 1 week ago

justinchuby commented 2 weeks ago 
<OpOverload(op='aten.__rshift__', overload='Scalar')>
<OpOverload(op='aten._adaptive_avg_pool2d', overload='default')>
<OpOverload(op='aten._cdist_forward', overload='default')>
<OpOverload(op='aten._convolution', overload='default')>
<OpOverload(op='aten._fake_quantize_per_tensor_affine_cachemask_tensor_qparams', overload='default')>
<OpOverload(op='aten._linalg_det', overload='default')>
<OpOverload(op='aten._linalg_slogdet', overload='default')>
<OpOverload(op='aten._prelu_kernel', overload='default')>
<OpOverload(op='aten.add', overload='Scalar')>
<OpOverload(op='aten.add', overload='Tensor')>
<OpOverload(op='aten.affine_grid_generator', overload='default')>
<OpOverload(op='aten.aminmax', overload='default')>
<OpOverload(op='aten.binary_cross_entropy_with_logits', overload='default')>
<OpOverload(op='aten.bitwise_and', overload='Tensor')>
<OpOverload(op='aten.bucketize', overload='Tensor')>
<OpOverload(op='aten.conv_tbc', overload='default')>
<OpOverload(op='aten.fake_quantize_per_tensor_affine_cachemask', overload='default')>
<OpOverload(op='aten.fill', overload='Scalar')>
<OpOverload(op='aten.fmod', overload='Scalar')>
<OpOverload(op='aten.fmod', overload='Tensor')>
<OpOverload(op='aten.glu', overload='default')>
<OpOverload(op='aten.grid_sampler_3d', overload='default')>
<OpOverload(op='aten.hann_window', overload='default')>
<OpOverload(op='aten.im2col', overload='default')>
<OpOverload(op='aten.index_add', overload='default')>
<OpOverload(op='aten.index_copy', overload='default')>
<OpOverload(op='aten.index_fill', overload='int_Scalar')>
<OpOverload(op='aten.index_put', overload='default')>
<OpOverload(op='aten.le', overload='Scalar')>
<OpOverload(op='aten.lerp', overload='Scalar')>
<OpOverload(op='aten.linalg_cross', overload='default')>
<OpOverload(op='aten.masked_scatter', overload='default')>
<OpOverload(op='aten.masked_select', overload='default')>
<OpOverload(op='aten.mv', overload='default')>
<OpOverload(op='aten.nan_to_num', overload='default')>
<OpOverload(op='aten.nll_loss2d_forward', overload='default')>
<OpOverload(op='aten.nll_loss_forward', overload='default')>
<OpOverload(op='aten.norm', overload='ScalarOpt_dim_dtype')>
<OpOverload(op='aten.pixel_unshuffle', overload='default')>
<OpOverload(op='aten.pow', overload='Scalar')>
<OpOverload(op='aten.prod', overload='dim_int')>
<OpOverload(op='aten.remainder', overload='Scalar')>
<OpOverload(op='aten.remainder', overload='Tensor')>
<OpOverload(op='aten.repeat_interleave', overload='Tensor')>
<OpOverload(op='aten.rsub', overload='Tensor')>
<OpOverload(op='aten.scatter', overload='src')>
<OpOverload(op='aten.scatter', overload='value')>
<OpOverload(op='aten.silu', overload='default')>
<OpOverload(op='aten.sort', overload='default')>
<OpOverload(op='aten.std', overload='correction')>
<OpOverload(op='aten.std_mean', overload='correction')>
<OpOverload(op='aten.sym_size', overload='int')>
<OpOverload(op='aten.take', overload='default')>
<OpOverload(op='aten.unsafe_split', overload='Tensor')>
<OpOverload(op='torchvision.nms', overload='default')>
<OpOverload(op='torchvision.roi_align', overload='default')>
<OpOverload(op='torchvision.roi_pool', overload='default')>