Construct an ExecutorTpu BaseRunner. Args: train_cfg: SingleTaskModelParams or MultiTaskModelParams ps_params_dict: A dict of top-level task name -> ProgramSchedule params, if train_cfg is a SingleTaskModelParams, we expect only one entry. *args: List args to pass thro
(self, train_cfg, ps_params_dict, *args, **kwargs)
source not stored for this graph (policy: none)
no test coverage detected