(op, ignore_existing=False)
| 5292 | # public API. There are still a few internal uses of colocate_with. Add internal |
| 5293 | # only API for those uses to avoid deprecation warning. |
| 5294 | def colocate_with(op, ignore_existing=False): |
| 5295 | return _colocate_with_for_gradient(op, None, ignore_existing=ignore_existing) |
| 5296 | |
| 5297 | |
| 5298 | @deprecation.deprecated( |
no test coverage detected