Searched refs:_get_value_per_replica (Results 1 – 1 of 1) sorted by relevance
2126 def _get_value_per_replica(tensor_spec_per_input): function2130 return nest.map_structure(_get_value_per_replica, tensor_spec)