code: def test_relay_cudnn_softmax(shape, axis, dtype):
      x = tvm.relay.var("x", tvm.relay.TensorType(shape, dtype))
      softmax = relay.op.nn.softmax(x, axis=axis)
      _verify_cudnn_relay(softmax)
  
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import te
  from tvm import relay
  from tvm.contrib import cudnn
  from tvm.contrib.nvcc import have_fp16
  from tvm.contrib import graph_executor
  import numpy
  import tvm.topi.testing
  import tvm.testing
  from tvm.relay.op.contrib.cudnn import partition_for_cudnn
label_desc: Use image node_id 14 for calling a global function with example usage: test_relay_cudnn_softmax(shape, axis, dtype) without return types