Skip to content

Conversation

@houseroad
Copy link
Member

@houseroad houseroad commented Jun 17, 2019

cosine_similarity has two non-tensor parameters, needs some special handling. Add the support for its export in this diff.

@pytorchbot pytorchbot added the module: onnx Related to torch.onnx label Jun 17, 2019
Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@houseroad houseroad force-pushed the cosine_similarity branch from ac7a795 to 5456c29 Compare June 17, 2019 23:20
Copy link
Contributor

@facebook-github-bot facebook-github-bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

@facebook-github-bot
Copy link
Contributor

@houseroad merged this pull request in f1c1d1a.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Merged module: onnx Related to torch.onnx

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants