Question on params['context']

[kdatta]

In the toy_model_tpu.py exampe, params['context'] is used to understand device assignments and host placements. Where is its value populated?

def model_fn(features, labels, mode, params): ... if FLAGS.use_tpu: ctx = params['context']