This is an automated email from the ASF dual-hosted git repository.
wuwei pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 603f8bd721 [Unity][Op] Negative Log Likelihood Loss (#14517)
add bb479e66dd [Unity][Transform] Some Improvements on pass DecomposeOps
(#14512)
No new revisions were added by this update.
Summary of changes:
include/tvm/relax/attrs/nn.h | 2 +
include/tvm/relax/transform.h | 23 +-
python/tvm/relax/op/nn/nn.py | 6 +-
python/tvm/relax/transform/legalize_ops/nn.py | 4 +
python/tvm/relax/transform/transform.py | 34 +-
src/relax/op/nn/nn.cc | 3 +-
src/relax/op/nn/nn.h | 2 +-
src/relax/transform/decompose_composite_ops.cc | 174 ------
src/relax/transform/decompose_ops.cc | 269 +++++++++
src/relax/transform/fold_constant.cc | 2 +-
tests/python/relax/test_op_nn.py | 6 +-
.../test_transform_decompose_composite_ops.py | 174 ------
tests/python/relax/test_transform_decompose_ops.py | 391 +++++++++++++
.../test_transform_legalize_ops_manipulate.py | 2 +-
.../python/relax/test_transform_legalize_ops_nn.py | 610 ++++++++++++++-------
15 files changed, 1148 insertions(+), 554 deletions(-)
delete mode 100644 src/relax/transform/decompose_composite_ops.cc
create mode 100644 src/relax/transform/decompose_ops.cc
delete mode 100644 tests/python/relax/test_transform_decompose_composite_ops.py
create mode 100644 tests/python/relax/test_transform_decompose_ops.py
