Searched refs:run_gpu_or_tpu (Results 1 – 2 of 2) sorted by relevance
599 @test_util.run_gpu_or_tpu610 @test_util.run_gpu_or_tpu643 @test_util.run_gpu_or_tpu660 @test_util.run_gpu_or_tpu
1772 def run_gpu_or_tpu(func=None): function