google / google/torchax

Convolution backward op missing

Open
#65 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
243
Forks
38
Avg merge
15h 32m
Merged PRs (30d)
2

Description

Observed when trying a training job that is doing convolution:
```
torchax/tensor.py", line 441, in _get_op_or_decomp
raise OperatorNotFound(
torchax.tensor.OperatorNotFound: Operator with name aten::convolution_backward has no lowering
```

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.