Searched refs:assignment_map (Results 1 – 2 of 2) sorted by relevance
220 def init_from_checkpoint(ckpt_dir_or_file, assignment_map): argument346 ckpt_dir_or_file, assignment_map)354 def _init_from_checkpoint(ckpt_dir_or_file, assignment_map): argument359 if isinstance(assignment_map, abc.Mapping):360 assignment_map = assignment_map.items()366 assignment_map, key=sort_key):
345 …argspec: "args=[\'ckpt_dir_or_file\', \'assignment_map\'], varargs=None, keywords=None, defaults=N…