tf.colocate_with

tf.colocate_with(
    op,
    ignore_existing=False
)

Defined in tensorflow/python/framework/ops.py.