spark/spark-tensorflow-distributor/spark_tensorflow_distributor/mirrored_strategy_runner.py (2 lines): - line 193: # TODO: handle the case that driver gpu resources - line 204: # TODO: Check GPU utilization to avoid resource contention