Searched refs:on_device_training_loop (Results 1 – 2 of 2) sorted by relevance
63 def on_device_training_loop(func): function
3175 @tpu_function.on_device_training_loop3200 @tpu_function.on_device_training_loop3223 @tpu_function.on_device_training_loop