Searched refs:is_input_sharded_per_core (Results 1 – 2 of 2) sorted by relevance
91 if self._internal_ctx.is_input_sharded_per_core():394 def is_input_sharded_per_core(self): member in _InternalTPUContext476 if self.is_input_sharded_per_core() or (
1206 self._sharded_per_core = ctx.is_input_sharded_per_core()