class_docstr: Microbenchmarks for the gather op.
class_docstr_tok: ["Microbenchmarks","for","the","gather","op","."]
code: def _benchmarkGather(self, name, axis, gather_indices, use_xla_jit):
      def BuilderFn():
          inputs = variables.Variable(
              array_ops.zeros(
                  [100, 100, 10, 100, 50], dtype=dtypes.float32
              ),
              dtype=dtypes.float32,
              name="input",
          )
          indices = variables.Variable(
              gather_indices, dtype=dtypes.int32, name="indices"
          )
          gather_t = array_ops.gather(inputs, indices, axis=axis)
          return "%s.axis%d" % (name, axis), [gather_t]
  
      xla_test.Benchmark(
          self, BuilderFn, use_xla_jit=use_xla_jit, device="cpu"
      )
  
imports: import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import flags
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the GatherBenchmark obj's underlying member method code with example usage: obj._benchmarkGather(name, axis, gather_indices, use_xla_jit) and returns: unknown,  list