New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[docs][ao] Add missing documentation for torch.quantized_batch_norm #63240
Conversation
Summary: Op is exposed via torch.quantized_batch_norm to the end user without any existing documentation Test Plan: CI Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 3f292c5 (more details on the Dr. CI page):
1 failure not recognized by patterns:
1 job timed out:
🚧 1 fixed upstream failure:These were probably caused by upstream breakages that were already fixed.
Please rebase on the
|
…atch_norm" Summary: Op is exposed via torch.quantized_batch_norm to the end user without any existing documentation Test Plan: CI Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
…atch_norm" Summary: Op is exposed via torch.quantized_batch_norm to the end user without any existing documentation Test Plan: CI Reviewers: Subscribers: Tasks: Tags: [ghstack-poisoned]
@supriyar has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
torch/_torch_docs.py
Outdated
|
||
Arguments: | ||
input (Tensor): quantized tensor | ||
weight (Tensor): tensor that corresponds to the gamma, size C |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Can we also specify that weight , bias and other tensors are in floating point?
…atch_norm" Summary: Op is exposed via torch.quantized_batch_norm to the end user without any existing documentation Test Plan: CI Reviewers: Subscribers: Tasks: Tags: Differential Revision: [D30316431](https://our.internmc.facebook.com/intern/diff/D30316431) [ghstack-poisoned]
@supriyar has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
This pull request has been merged in a090073. |
Stack from ghstack:
Summary:
Op is exposed via torch.quantized_batch_norm to the end user without any existing documentation
Test Plan:
CI
Reviewers:
Subscribers:
Tasks:
Tags:
Differential Revision: D30316431