Searched refs:under_tpu_inference_context (Results 1 – 3 of 3) sorted by relevance
510 if tpu.under_tpu_inference_context():546 if tpu.under_tpu_inference_context():565 if tpu.under_tpu_inference_context():592 if tpu.under_tpu_inference_context():705 if tpu.under_tpu_inference_context():730 if tpu.under_tpu_inference_context():885 if (tpu.under_tpu_inference_context() and891 if not tpu.under_tpu_inference_context():956 if tpu.under_tpu_inference_context():994 if tpu.under_tpu_inference_context():[all …]
454 if tpu.under_tpu_inference_context():478 if tpu.under_tpu_inference_context():593 if tpu.under_tpu_inference_context():616 if tpu.under_tpu_inference_context():
2085 def under_tpu_inference_context() -> bool: function