[RELAY] CompileEngine update, nn conv2d, fix dense, pool. (#2082)
Showing
- include/tvm/relay/op_attr_types.h 2 additions, 1 deletioninclude/tvm/relay/op_attr_types.h
- python/tvm/relay/op/_tensor.py 2 additions, 7 deletionspython/tvm/relay/op/_tensor.py
- python/tvm/relay/op/nn/__init__.py 1 addition, 0 deletionspython/tvm/relay/op/nn/__init__.py
- python/tvm/relay/op/nn/_nn.py 167 additions, 9 deletionspython/tvm/relay/op/nn/_nn.py
- python/tvm/relay/op/op.py 9 additions, 0 deletionspython/tvm/relay/op/op.py
- src/relay/backend/compile_engine.cc 6 additions, 1 deletionsrc/relay/backend/compile_engine.cc
- src/relay/op/nn/nn.cc 2 additions, 3 deletionssrc/relay/op/nn/nn.cc
- src/relay/op/nn/pooling.cc 75 additions, 6 deletionssrc/relay/op/nn/pooling.cc
- tests/python/relay/test_backend_interpreter.py 0 additions, 22 deletionstests/python/relay/test_backend_interpreter.py
- tests/python/relay/test_op_level1.py 60 additions, 2 deletionstests/python/relay/test_op_level1.py
- tests/python/relay/test_op_level2.py 160 additions, 34 deletionstests/python/relay/test_op_level2.py
- topi/python/topi/util.py 5 additions, 4 deletionstopi/python/topi/util.py
Loading
Please register or sign in to comment