Home
last modified time | relevance | path

Searched refs:get_collection_ref (Results 1 – 25 of 41) sorted by relevance

12

/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py53 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
78 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
114 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
127 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
153 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
257 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
286 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
323 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
358 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
387 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
[all …]
Dcluster_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
56 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
74 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
94 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
121 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
147 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Ditem_test.py55 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
68 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
90 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
118 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dmodel_analyzer_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
60 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dcost_analyzer_test.py48 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
73 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
143 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dgraph_placer_test.py57 train_op = g.get_collection_ref(tf_ops.GraphKeys.TRAIN_OP)
106 train_op = tf_ops.get_collection_ref(tf_ops.GraphKeys.TRAIN_OP)
Dmemory_optimizer_test.py48 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
73 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dtf_optimizer_test.py45 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
/external/tensorflow/tensorflow/python/ops/
Dtemplate.py308 ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES))
310 ops.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES))
317 trainable_variables = ops.get_collection_ref(
331 variables = ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES)
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Drnn_cells.py115 if self._mask not in ops.get_collection_ref(core_layers.MASK_COLLECTION):
264 if self._mask not in ops.get_collection_ref(core_layers.MASK_COLLECTION):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_gradient.py87 table_gradients = g.get_collection_ref(
D_tpu_estimator_embedding.py83 collection = graph.get_collection_ref(tpu_fc._TPU_FC_TO_SCOPE) # pylint: disable=protected-access
Dfeature_column.py405 collection = g.get_collection_ref(_TPU_FC_TO_SCOPE)
/external/tensorflow/tensorflow/python/framework/
Dframework_lib.py46 from tensorflow.python.framework.ops import get_collection_ref
/external/tensorflow/tensorflow/contrib/opt/python/training/
Delastic_average_optimizer.py302 var = g.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES)[idx]
306 del g.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES)[idx]
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py169 table_gradients = g.get_collection_ref(
/external/tensorflow/tensorflow/contrib/training/python/training/
Dtraining.py469 train_ops = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-graph.pbtxt90 name: "get_collection_ref"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-graph.pbtxt90 name: "get_collection_ref"
/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py620 save_graph.get_collection_ref("partvar").append(v0)
632 v0, = save_graph.get_collection_ref("partvar")
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model.py270 g.get_collection_ref(ops.GraphKeys.UPDATE_OPS).extend(clone.state_updates)
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
Dkeras_tpu_variables.py289 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
/external/tensorflow/tensorflow/python/training/
Doptimizer.py613 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
708 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py143 mutable_collection = ops.get_collection_ref(collection_name)
/external/tensorflow/tensorflow/python/layers/
Dbase.py575 collection = ops.get_collection_ref(name)

12