Skip to content

Instantly share code, notes, and snippets.

@AmosLewis
Created January 26, 2023 05:30
Show Gist options
  • Save AmosLewis/fb4ea4f2880457771a034889ac905267 to your computer and use it in GitHub Desktop.
Save AmosLewis/fb4ea4f2880457771a034889ac905267 to your computer and use it in GitHub Desktop.
func.func @torch.aten.to.dtype(%arg0: !torch.vtensor<[1,128],i1>) -> !torch.vtensor<[1,128],si64> {
%int4 = torch.constant.int 4
%none = torch.constant.none
%false = torch.constant.bool false
%0 = torch.aten.to.dtype %arg0, %int4, %false, %false, %none : !torch.vtensor<[1,128],i1>, !torch.int, !torch.bool, !torch.bool, !torch.none -> !torch.vtensor<[1,128],si64>
return %0 : !torch.vtensor<[1,128],si64>
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment