| /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
|
