Lines Matching refs:num_shards
45 'num_shards',
58 num_shards: (Deprecated, ignored by TPUEstimator).
62 product(computation_shape) * num_shards.
91 num_shards=None,
101 # Check num_shards.
102 if num_shards is not None:
103 util_lib.check_positive_integer(num_shards, 'TPUConfig num_shards')
129 num_shards=num_shards,
