(device_type, op)
| 285 | |
| 286 | @staticmethod |
| 287 | def device_mismatch(device_type, op): |
| 288 | if device_type == _DEVICE_TYPE_TPU: |
| 289 | # pylint: disable=protected-access |
| 290 | return tpu._TPU_REPLICATE_ATTR not in op.node_def.attr |
| 291 | # pylint: enable=protected-access |
| 292 | return False |
| 293 | |
| 294 | @staticmethod |
| 295 | def unsafe_scalar_trace(op): |