Searched refs:is_tpu_strat (Results 1 – 1 of 1) sorted by relevance
215 is_tpu_strat = lambda k: k.__name__.startswith("TPUStrategy") function217 return is_tpu_strat(clz) or any(map(is_tpu_strat, clz.__bases__))