Searched refs:_rebatch_as_dynamic (Results 1 – 1 of 1) sorted by relevance
911 _rebatch_as_dynamic, self._element_spec, expand_composites=False)1140 _rebatch_as_dynamic, self._element_spec, expand_composites=False)1318 _rebatch_as_dynamic, self._element_spec, expand_composites=False)2198 def _rebatch_as_dynamic(per_replica_spec): function