saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add comment to explain local scalar dense (#12179)

Browse source 
* add comment to explain local scalar dense

* spacing
This commit is contained in:
msftlincoln 2022-07-15 09:03:43 -04:00 committed by GitHub
parent 05c31a036d
commit fafb24142f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/eager/opgen/opgen/atenops.py
View file

@ -159,7 +159,9 @@ hand_implemented = {
"aten::eq.Scalar_out": Cast(Equal("self", "other"), to="GetONNXTensorProtoDataType(out.scalar_type())"),
"aten::bitwise_and.Tensor_out": MakeTorchFallback(),
"aten::masked_select": GatherND("self", Transpose(NonZero(Expand("mask", Shape("self"))))),
"aten::_local_scalar_dense": MakeTorchFallback(),
"aten::_local_scalar_dense": MakeTorchFallback(), # This function extracts a scalar value from
# a tensor with exactly one value; there's no need to try to do this on an ORT device.
# See CPU impl at pytorch/blob/master/aten/src/ATen/native/Scalar.cpp
"aten::gt.Scalar_out": MakeTorchFallback(),
"aten::lt.Scalar_out": MakeTorchFallback(),
"aten::equal": SignatureOnly(),