Searched refs:is_input_broadcast_with_iterators (Results 1 – 2 of 2) sorted by relevance
95 elif self._internal_ctx.is_input_broadcast_with_iterators():406 def is_input_broadcast_with_iterators(self): member in _InternalTPUContext472 if (self.is_running_on_cpu() or self.is_input_broadcast_with_iterators()):487 if (self.is_running_on_cpu() or self.is_input_broadcast_with_iterators()):636 not self.is_input_broadcast_with_iterators()):647 not self.is_input_broadcast_with_iterators()):651 if num_hosts > 1 and not self.is_input_broadcast_with_iterators():662 not self.is_input_broadcast_with_iterators()):666 if num_hosts > 1 and not self.is_input_broadcast_with_iterators():
222 if not ctx.is_input_broadcast_with_iterators():1267 elif self._ctx.is_input_broadcast_with_iterators():