Searched refs:value_specs (Results 1 – 2 of 2) sorted by relevance
2127 value_specs = [tensor_spec_per_input for _ in range(num_replicas)]2128 return values.PerReplicaSpec(*value_specs)
383 def __init__(self, *value_specs): argument384 self._value_specs = tuple(value_specs)