code: def test_tfn_argmax(self):
      @def_function.function
      def wrapped_fn(x):
          return math_ops.argmax(x, axis=2)
  
      def fn():
          wrapped_fn([0, 1])
  
      self.assert_trace_line_count(fn, count=15, filtering_enabled=True)
      self.assert_trace_line_count(
          fn, count=25, filtering_enabled=False
      )
  
file_path: tensorflow/tensorflow/python/util/traceback_utils_test.py
imports: import traceback
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.util import traceback_utils
label_desc: Use image node_id 6 for calling the TracebackUtilsTest obj's underlying member method code with example usage: obj.test_tfn_argmax() and returns: math_ops