File size: 125 Bytes
86b1a7e |
1 2 3 4 5 |
from enum import Enum
def execute_graph() -> None:
if _acceleration_type == AccelerationType.TPU:
xm.mark_step() |
86b1a7e |
1 2 3 4 5 |
from enum import Enum
def execute_graph() -> None:
if _acceleration_type == AccelerationType.TPU:
xm.mark_step() |