/external/tensorflow/tensorflow/contrib/data/python/ops/ |
iterator_ops.py | 76 with ops.colocate_with(self.op):
|
/external/tensorflow/tensorflow/contrib/opt/python/training/ |
variable_clipping_optimizer.py | 116 with ops.colocate_with(var): 128 with ops.colocate_with(var): 135 with ops.colocate_with(var): 142 with ops.colocate_with(var):
|
drop_stale_gradient_optimizer.py | 65 with ops.colocate_with(loss): 81 with ops.colocate_with(global_step): 103 with ops.control_dependencies(gradients), ops.colocate_with(global_step):
|
/external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
moving_stats.py | 78 with ops.colocate_with(variance_var): 79 with ops.colocate_with(mean_var): 154 with ops.colocate_with(log_mean_exp_var):
|
/external/tensorflow/tensorflow/contrib/lookup/ |
lookup_ops.py | 380 with ops.colocate_with(self._table_ref): 408 with ops.colocate_with(self._table_ref): 438 with ops.colocate_with(self._table_ref): 456 with ops.colocate_with(self._table_ref): 479 with ops.colocate_with(self.op._table_ref): 582 with ops.colocate_with(self._table_ref): 609 with ops.colocate_with(self._table_ref): 642 with ops.colocate_with(self._table_ref): 660 with ops.colocate_with(self._table_ref): 683 with ops.colocate_with(self.op._table_ref) [all...] |
/external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
clustering_ops.py | 191 with ops.colocate_with(inp, ignore_existing=True): 221 with ops.colocate_with(clusters, ignore_existing=True): 224 with ops.colocate_with(inp, ignore_existing=True): 256 with ops.colocate_with(clusters, ignore_existing=True): 259 with ops.colocate_with(inp, ignore_existing=True): 326 with ops.colocate_with(inp, ignore_existing=True): 402 with ops.colocate_with(update_in_steps, ignore_existing=True): 415 with ops.colocate_with( 422 with ops.colocate_with(cluster_centers_var, ignore_existing=True): 425 with ops.colocate_with(None, ignore_existing=True) [all...] |
factorization_ops.py | 442 with ops.colocate_with(f): 445 with ops.colocate_with(gramian): 482 with ops.colocate_with(cache): 494 with ops.colocate_with(c): 631 with ops.colocate_with(factor[0]): [all...] |
gmm_ops.py | 201 with ops.colocate_with(self._cluster_centers_initialized): 206 with ops.colocate_with(self._means): 214 with ops.colocate_with(self._covs):
|
/external/tensorflow/tensorflow/python/training/ |
slot_creator.py | 112 with ops.colocate_with(primary): 145 with ops.colocate_with(primary):
|
adam.py | 124 colocate_with=first_var) 127 colocate_with=first_var) 216 with ops.colocate_with(beta1_power):
|
adagrad_da.py | 89 with ops.colocate_with(v): 105 with ops.colocate_with(self._learning_rate_tensor):
|
adagrad.py | 66 with ops.colocate_with(v):
|
proximal_adagrad.py | 72 with ops.colocate_with(v):
|
moving_averages.py | 81 with ops.colocate_with(variable): 190 with ops.colocate_with(unbiased_var):
|
optimizer.py | 556 with ops.name_scope("update_" + scope_name), ops.colocate_with(var): 562 with ops.colocate_with(global_step): 643 def _create_non_slot_variable(self, initial_value, name, colocate_with): 646 graph = colocate_with.graph 653 with ops.colocate_with(colocate_with): [all...] |
/external/tensorflow/tensorflow/python/ops/ |
numerics.py | 44 with ops.colocate_with(t):
|
embedding_ops.py | 150 with ops.colocate_with(params[0]): 183 with ops.colocate_with(params[p]): 214 with ops.colocate_with(params[p]): 240 with ops.colocate_with(params[0]):
|
clip_ops.py | 160 with ops.colocate_with(v): 243 with ops.colocate_with(v):
|
tensor_array_ops.py | 161 with ops.device(None), ops.colocate_with(None, ignore_existing=True): 216 with ops.colocate_with(self._colocate_with[0]): 239 with ops.colocate_with(self._handle): 289 with ops.colocate_with(self._handle): 835 def _colocate_with(self, colocate_with): 836 self._implementation._colocate_with = colocate_with [all...] |
/external/tensorflow/tensorflow/contrib/all_reduce/python/ |
all_reduce.py | 54 with ops.colocate_with(t): 73 with ops.colocate_with(t): 99 with ops.colocate_with(tensor): 155 with ops.colocate_with(t): 181 with ops.colocate_with(tensor): 372 with ops.colocate_with(x[0]): 404 with ops.colocate_with(chunks_by_dev[d][0]): 421 with ops.colocate_with(x[0]): 609 with ops.colocate_with(input_tensors[d]): 703 with ops.colocate_with(t) [all...] |
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
embedding_ops.py | 730 with ops.colocate_with(params[0]): [all...] |
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/ |
model.py | 107 [update_op]), (ops.colocate_with(global_step)):
|
/external/tensorflow/tensorflow/contrib/input_pipeline/python/ops/ |
input_pipeline_ops.py | 94 with ops.colocate_with(counter):
|
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
math_utils.py | [all...] |
/external/tensorflow/tensorflow/python/framework/ |
framework_lib.py | 35 @@colocate_with 89 from tensorflow.python.framework.ops import colocate_with
|