class_docstr: A context object holding state about the TPU computation being built.
code: def set_number_of_shards(self, number_of_shards):
      self._number_of_shards = number_of_shards
  
file_path: tensorflow-upstream/tensorflow/python/tpu/tpu_function.py
imports: import contextlib
  import threading
label_desc: Use image node_id 3 for calling the TpuContext obj's underlying member method code with example usage: obj.set_number_of_shards(number_of_shards) without return types