Searched defs:shard_info (Results 1 – 1 of 1) sorted by relevance
76 def __call__(self, shape=None, dtype=None, shard_info=None): argument102 def __init__(self, checkpoint_position, shape=None, shard_info=None): argument