docs(aten::zeros): Add some comments

narendasan · narendasan · commit 0f783dadfb23 · 2021-02-23T15:37:41.000-08:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/core/conversion/evaluators/aten.cpp b/core/conversion/evaluators/aten.cpp
@@ -120,6 +120,7 @@ auto aten_registrations TRTORCH_UNUSED =
                     [](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {
                       auto options = torch::TensorOptions().layout(torch::kStrided).device(torch::kCUDA);
 
+                      // Input 1 here is the dtype
                       if (!args.at(n->input(1)).isNone() && !args.at(n->input(1)).IValue()->isNone()) {
                         options = options.dtype(c10::ScalarType(args.at(n->input(1)).unwrapToInt()));
                       }

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ auto aten_registrations TRTORCH_UNUSED =`
`120`	`120`	`[](const torch::jit::Node* n, kwargs& args) -> c10::optional<torch::jit::IValue> {`
`121`	`121`	`auto options = torch::TensorOptions().layout(torch::kStrided).device(torch::kCUDA);`
`122`	`122`
	`123`	`+ // Input 1 here is the dtype`
`123`	`124`	`if (!args.at(n->input(1)).isNone() && !args.at(n->input(1)).IValue()->isNone()) {`
`124`	`125`	`options = options.dtype(c10::ScalarType(args.at(n->input(1)).unwrapToInt()));`
`125`	`126`	`}`