Searched defs:colocate_with (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
H A Dframework_lib.py35 @@colocate_with
89 from tensorflow.python.framework.ops import colocate_with namespace
H A Dops.py2740 # Stack of colocate_with ops
4032 def colocate_with(self, op, ignore_existing=False): member in class:Graph
4041 with g.colocate_with(a):
4076 # By default, colocate_with resets the device function stack,
4077 # since colocate_with is typically used in specific internal
4721 @tf_export("colocate_with")
4722 def colocate_with(op, ignore_existing=False): function
4724 return get_default_graph().colocate_with(op, ignore_existing)

Completed in 227 milliseconds