code: def test_max_grad(executor_kind):
      verify_max_grad(executor_kind, (10, 10), axis=None)
      verify_max_grad(executor_kind, (10, 10), axis=-1)
      verify_max_grad(
          executor_kind, (6, 3, 2), axis=(1, 2), keepdims=True
      )
      verify_max_grad(
          executor_kind, (5, 4, 3), axis=(0, 2), exclude=True
      )
  
file_path: tvm/tests/python/relay/test_op_grad_level4.py
imports: import pytest
  import numpy
  import tvm.testing
  from tvm import relay
  from tvm.relay.testing import check_grad, _np_randn_from_type
label_desc: Use image node_id 4 for calling a global function with example usage: test_max_grad(executor_kind) without return types