[torchlib] Implement missing operators uncovered by torch.onnx tests

[justinchuby]

NOTE: Some of these may just be incorrectly registered. In this case update the @torch_op decorator call in the corresponding functions to register them correctly. It's usually just correcting the overload names. "default" overload can be omitted in names. E.g.

aten.add.Scalar -> @torch_op("aten::add.Scalar")
aten.glu.default -> @torch_op("aten::glu")

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

[justinchuby]

- func: type_as(Tensor self, Tensor other) -> Tensor
  variants: method

• [torchlib] Implement aten::type_as #1726

[justinchuby]

<OpOverload(op='aten.group_norm', overload='default')