Searched refs:input_contexts (Results 1 – 4 of 4) sorted by relevance
44 input_contexts = []46 input_contexts.append(53 input_contexts)
239 def __init__(self, input_fn, input_workers, input_contexts): argument255 if input_workers.num_workers != len(input_contexts):259 (input_workers.num_workers, len(input_contexts)))262 for i, ctx in enumerate(input_contexts):
266 input_contexts = []269 input_contexts.append(distribute_lib.InputContext(274 input_fn, self._input_workers, input_contexts)
577 input_contexts = []580 input_contexts.append(distribute_lib.InputContext(585 input_fn, self._input_workers, input_contexts)