[torchlib] Implement various upsample functions

- [x] aten_upsample_bicubic2d https://github.com/microsoft/onnxscript/pull/1208
- [x] aten__upsample_bicubic2d_aa #1259 
- [x] aten__upsample_bilinear2d_aa #1259 
- [x] aten_upsample_bilinear2d.vec https://github.com/microsoft/onnxscript/pull/1249
- [x] aten_upsample_linear1d #1245 
- [x] aten_upsample_nearest1d #1246 
- [x] aten_upsample_nearest3d #1246 
- [x] aten_upsample_trilinear3d  #1260 

Reference: https://github.com/microsoft/onnxscript/blob/24d77d96ee343fd6cac4b253a6c8e10c7ac4bd2a/onnxscript/function_libs/torch_lib/ops/nn.py#L2121-L2137

	@torch_op("aten::upsample_bilinear2d", trace_only=True)
	def aten_upsample_bilinear2d(
	self: TReal,
	output_size: Optional[INT64] = None,
	scales_h: Optional[float] = None,
	scales_w: Optional[float] = None,
	align_corners: bool = True, # pylint: disable=unused-argument
	) -> TReal:
	"""upsample_bilinear2d(Tensor self, SymInt[2] output_size, bool align_corners, float? scales_h=None, float? scales_w=None) -> Tensor"""

	if output_size is not None:
	result = _aten_upsample_bilinear2d_output_size(self, output_size)
	else:
	assert scales_h is not None
	assert scales_h == scales_w
	result = _aten_upsample_bilinear2d_scales(self, scales_h, scales_w)
	return result

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[torchlib] Implement various upsample functions #1159

justinchuby
openedon Nov 16, 2023

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

[torchlib] Implement various upsample functions #1159

Description

justinchubyopenedon Nov 16, 2023

Metadata