Searched refs:PerReplicaSpec (Results 1 – 6 of 6) sorted by relevance
372 values.PerReplicaSpec(378 values.PerReplicaSpec(384 values.PerReplicaSpec(
2128 return values.PerReplicaSpec(*value_specs)2135 if isinstance(spec, values.PerReplicaSpec):2200 assert isinstance(per_replica_spec, values.PerReplicaSpec), per_replica_spec2211 return values.PerReplicaSpec(
367 return PerReplicaSpec(376 class PerReplicaSpec(type_spec.TypeSpec): class
920 if isinstance(element_spec, values.PerReplicaSpec):
136 PER_REPLICA_SPEC = 8; // PerReplicaSpec from distribute/values.py
514 values.PerReplicaSpec,