You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Makes it easy to declare shape helper classes of ONNX operators and reduce boilerplate code.
As an example, to declare shape helpers for the ONNXArgMinOp & ONNXArgMaxOp:
DECLARE_SHAPE_HELPER(ONNXArgMaxOp)
DECLARE_SHAPE_HELPER(ONNXArgMinOp)
Implement shape inference for the ArgMinOp ONNX operator.
Signed-off-by: Ettore Tiotto <etiotto@ca.ibm.com>
Before:
2hr 15m https://github.com/llvm/torch-mlir/actions/runs/2955716430
after:
varies but sometimes times out at 6hr. and may pass on a re-run.
The text was updated successfully, but these errors were encountered: