-
Notifications
You must be signed in to change notification settings - Fork 3.6k
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
adding tests for cast operation #543
Conversation
@bddppq , @houseroad : please feel free to review this PR instead of earlier PR on cast operator tests. |
✅ Build onnx 0.3.1290 completed (commit 75cb369816 by @houseroad) |
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.
LG, one comment to address.
Thanks!
onnx/backend/test/case/node/cast.py
Outdated
for case in test_cases: | ||
from_type = case[0] | ||
to_type = case[1] | ||
input = np.random.random_sample(shape).astype(type_map[getattr(TensorProto, from_type)]) |
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.
In the generated doc, user may be confused about the type_map. If we put TENSOR_TYPE_TO_NP_TYPE here directly, it will be clear to users.
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.
Sure, pushed an update.
@onnxbot test this please |
* adding tests for cast operation * addressing PR feedback
* adding tests for cast operation * addressing PR feedback
No description provided.