TPU 训练支持
作者: Borda创建于 2026年5月22日更新于 2026年8月23日
标签enhancement
Task 1.1 — XLA branch in `_bilinear_grid_sample` (`tensors.py:228`): `F.grid_sample` backward is a known CPU-fallback op on XLA. The gather-based path already used for MPS is fully XLA-compatible. One-line fix:
内容来源: roboflow/rf-detr