[Runtime][Contrib] Support cudnn softmax (#5214)
authorHaichen Shen <shenhaichen@gmail.com>
Mon, 6 Apr 2020 03:53:59 +0000 (20:53 -0700)
committerGitHub <noreply@github.com>
Mon, 6 Apr 2020 03:53:59 +0000 (20:53 -0700)
commit799ff356b9308da9c06b61c214508851a35c3b93
tree493a80e830af212f541134375ccab6ffd681f309
parent02eb18332a4f9b9491da0d4e8ed01c49fec2ac5e
[Runtime][Contrib] Support cudnn softmax (#5214)
16 files changed:
python/tvm/contrib/cudnn.py
python/tvm/relay/op/nn/_nn.py
python/tvm/relay/op/op_attrs.py
python/tvm/relay/op/strategy/cuda.py
python/tvm/relay/op/strategy/generic.py
python/tvm/relay/op/strategy/hls.py
python/tvm/relay/op/strategy/opengl.py
python/tvm/relay/op/strategy/x86.py
src/relay/op/nn/nn.cc
src/runtime/contrib/cudnn/cudnn_utils.cc
src/runtime/contrib/cudnn/cudnn_utils.h
src/runtime/contrib/cudnn/softmax.cc [new file with mode: 0644]
tests/python/contrib/test_cudnn.py
topi/python/topi/cuda/__init__.py
topi/python/topi/cuda/softmax.py
topi/python/topi/nn/softmax.py