This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 70183ed [Docker] Fix ordering of tf and tflite installs in ci_qemu
(#8315)
add 2186835 [Relay, TOPI] Add negative log likelihood loss (nll_loss) op
(#8056)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 13 +++++
include/tvm/topi/nn.h | 48 ++++++++++++++++
python/tvm/relay/frontend/pytorch.py | 16 ++++++
python/tvm/relay/op/nn/_nn.py | 11 ++++
python/tvm/relay/op/nn/nn.py | 36 ++++++++++++
python/tvm/relay/op/op_attrs.py | 5 ++
python/tvm/topi/nn/__init__.py | 1 +
python/tvm/topi/nn/loss.py | 60 ++++++++++++++++++++
python/tvm/topi/testing/__init__.py | 1 +
python/tvm/topi/testing/nll_loss.py | 72 ++++++++++++++++++++++++
src/relay/op/nn/nn.cc | 80 +++++++++++++++++++++++++++
src/topi/nn.cc | 4 ++
tests/python/frontend/pytorch/test_forward.py | 24 ++++++++
tests/python/relay/test_op_level10.py | 41 ++++++++++++++
tests/python/topi/python/test_topi_loss.py | 70 +++++++++++++++++++++++
15 files changed, 482 insertions(+)
create mode 100644 python/tvm/topi/nn/loss.py
create mode 100644 python/tvm/topi/testing/nll_loss.py
create mode 100644 tests/python/topi/python/test_topi_loss.py
