Searched refs:tensor_spec_per_input (Results 1 – 1 of 1) sorted by relevance
2078 def _get_value_per_replica(tensor_spec_per_input): argument2079 value_specs = [tensor_spec_per_input for _ in range(num_replicas)]