masahi 0d38a92bcf [Relay, ONNX] Support gather_nd batch_dims attribute for TF/ONNX (#8084) ...

* Add GatherND batch_dim support

* adding tests

* test working

* improved reference code

* refactor ref func

* batch dim 2 tests from tf all passed

* batch_dim -> batch_dims

* add example

* minor change

* add onnx test

* fix onnx version

* fix lint

* remove move on batch_dims

* fix pylint

* fix compiler warning

* add shape constraint for batch_dim and update doc

* make the output shape doc clearer

2021-05-21 13:50:49 +09:00

..

tvm

[Relay, ONNX] Support gather_nd batch_dims attribute for TF/ONNX (#8084)

2021-05-21 13:50:49 +09:00

.gitignore

Generate requirements.txt from Python spec (#7289)

2021-02-04 06:51:12 -08:00

gen_requirements.py

Generate requirements.txt from Python spec (#7289)

2021-02-04 06:51:12 -08:00

setup.py

[RUNTIME] Add libbacktrace for backtraces with line numbers (#7153)

2021-03-18 09:08:08 -04:00