Skip to content

Instantly share code, notes, and snippets.

@shunting314
Created March 29, 2022 19:37
Show Gist options
  • Save shunting314/eabdf6c769c59bc384469717b8f9bb7f to your computer and use it in GitHub Desktop.
Save shunting314/eabdf6c769c59bc384469717b8f9bb7f to your computer and use it in GitHub Desktop.
One of the Fx graph generated by dynamo for yolov3: the graph create a tensor without specifying device and return it
graph():
%tensor : [#users=1] = call_function[target=torch.tensor](args = ((16, 12),), kwargs = {dtype: torch.float32})
return (tensor,)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment