Home
last modified time | relevance | path

Searched refs:train_op (Results 1 – 25 of 147) sorted by relevance

123456

/external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/
Dsdca_ops_test.py265 train_op = lr.minimize()
267 train_op.run()
268 lr.update_weights(train_op).run()
313 train_op = lr.minimize()
315 train_op.run()
316 lr.update_weights(train_op).run()
375 train_op = lr.minimize()
377 train_op.run()
378 lr.update_weights(train_op).run()
411 train_op = lr.minimize()
[all …]
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py250 train_op = constant_op.constant(3)
258 mon_sess.run(train_op)
268 train_op = constant_op.constant(3)
274 mon_sess.run(train_op)
279 mon_sess.run(train_op)
282 mon_sess.run(train_op)
287 mon_sess.run(train_op)
315 train_op = constant_op.constant(3)
321 mon_sess.run(train_op)
328 train_op = constant_op.constant(3)
[all …]
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Dlearning_test.py254 train_op = learning.create_train_op(total_loss, optimizer)
257 train_op, logdir, number_of_steps=300, log_every_n_steps=10)
289 train_op = learning.create_train_op(total_loss, optimizer)
304 sess.run([train_op])
323 train_op = learning.create_train_op(total_loss, optimizer, update_ops=[])
338 sess.run([train_op])
356 train_op = learning.create_train_op(total_loss, optimizer)
365 sess.run([train_op])
381 train_op = learning.create_train_op(
391 sess.run([train_op])
[all …]
/external/tensorflow/tensorflow/contrib/training/python/training/
Dtraining_test.py101 train_op = training.create_train_op(loss, optimizer)
104 self.assertTrue(train_op in ops.get_collection(ops.GraphKeys.TRAIN_OP))
119 train_op = training.create_train_op(loss, optimizer)
134 session.run(train_op)
152 train_op = training.create_train_op(loss, optimizer, update_ops=[])
167 session.run(train_op)
185 train_op = training.create_train_op(loss, optimizer)
194 session.run(train_op)
208 train_op = training.create_train_op(loss, optimizer, global_step=None)
217 session.run(train_op)
[all …]
Dtraining.py466 train_op = control_flow_ops.with_dependencies([grad_updates], total_loss)
470 if train_op not in train_ops:
471 train_ops.append(train_op)
473 return train_op
476 def train(train_op, argument
545 loss = session.run(train_op, run_metadata=run_metadata)
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py53 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
54 train_op.append(get_next)
78 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
79 train_op.append(get_next)
114 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
115 train_op.append(get_next)
127 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
128 train_op.append(get_next)
153 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
154 train_op.append(get_next)
[all …]
Dcluster_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
40 train_op.append(c)
56 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
57 train_op.append(c)
74 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
75 train_op.append(c)
94 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
95 train_op.append(c)
121 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
122 train_op.append(c)
[all …]
Ditem_test.py55 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
56 train_op.append(c)
68 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
69 train_op.append(c)
90 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
91 train_op.append(c)
118 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
119 train_op.append(a)
Dmemory_optimizer_test.py48 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
49 train_op.append(d)
73 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
74 train_op.append(d)
130 train_op = optimizer.minimize(loss)
133 return (metagraph, init_op.name, train_op.name, loss.name)
235 train_op = graph.get_operation_by_name(train_op_name)
239 self.evaluate(train_op)
240 self.evaluate(train_op)
287 train_op = optimizer.minimize(loss)
[all …]
Dmodel_analyzer_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
40 train_op.append(d)
60 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
61 train_op.append(c)
Dcost_analyzer_test.py48 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
49 train_op.append(d)
73 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
74 train_op.append(d)
143 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
144 train_op.append(d)
/external/tensorflow/tensorflow/contrib/learn/python/learn/
Dgraph_actions_test.py473 train_op = constant_op.constant(1.0)
477 g, output_dir=None, train_op=train_op, loss_op=loss_op)
482 train_op=constant_op.constant(1.0),
486 g, output_dir=self._output_dir, train_op=None, loss_op=loss_op)
491 train_op=constant_op.constant(1.0),
497 train_op=constant_op.constant(1.0),
508 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
514 train_op=train_op,
527 train_op = state_ops.assign_add(variables_lib.get_global_step(), 1)
531 train_op=train_op,
[all …]
Dmonitors_test.py624 self.train_op = state_ops.assign_add(self.global_step, 1)
629 def _run(self, monitor, step, train_op, sess): argument
631 sess.run(train_op)
651 self._run(monitor, 1, self.train_op, sess)
665 self._run(monitor, 1, self.train_op, sess)
666 self._run(monitor, 2, self.train_op, sess)
672 self._run(monitor, 3, self.train_op, sess)
677 self._run(monitor, 4, self.train_op, sess)
678 self._run(monitor, 5, self.train_op, sess)
684 self._run(monitor, 6, self.train_op, sess)
[all …]
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Dmodel_fn.py90 train_op=None, argument
146 get_graph_from_inputs((predictions, loss, train_op))
149 if train_op is None:
152 elif not isinstance(train_op, ops.Operation):
154 train_op = ops.convert_to_tensor(train_op).op
204 train_op=train_op,
302 train_op=self.train_op,
Destimator_input_test.py114 train_op = optimizers.optimize_loss(
119 return prediction, loss, train_op
130 train_op = optimizers.optimize_loss(
135 return prediction, loss, train_op
143 train_op = optimizers.optimize_loss(
149 mode=mode, predictions=prediction, loss=loss, train_op=train_op)
157 train_op = optimizers.optimize_loss(
165 }, loss, train_op
/external/tensorflow/tensorflow/contrib/constrained_optimization/python/
Dconstrained_optimizer.py138 train_op = train_op_callback()
140 train_op = train_op_callback()
142 return train_op
191 train_op = train_op_callback()
193 train_op = train_op_callback()
195 return train_op
/external/tensorflow/tensorflow/python/saved_model/
Dbuilder_impl.py232 train_op=None, argument
277 _add_op_to_signature_def_map(signature_def_map, train_op,
305 train_op=None, argument
355 _add_op_to_signature_def_map(signature_def_map, train_op,
429 def _add_collections(self, assets_collection, main_op, train_op): argument
436 self._add_train_op(train_op)
484 def _add_train_op(self, train_op): argument
497 if train_op is not None:
498 if (not isinstance(train_op, ops.Tensor) and
499 not isinstance(train_op, ops.Operation)):
[all …]
/external/tensorflow/tensorflow/contrib/boosted_trees/python/training/functions/
Dgbdt_batch_test.py259 train_op = gbdt_model.train(
269 train_op.run()
283 train_op.run()
352 train_op = gbdt_model.train(
362 train_op.run()
372 train_op.run()
407 train_op.run()
523 train_op = gbdt_model.train(
533 train_op.run()
545 train_op.run()
[all …]
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dlearning.py61 def train(train_op, argument
158 def train_step_with_pruning_fn(sess, train_op, global_step, argument
160 total_loss, should_stop = train_step_fn(sess, train_op, global_step,
166 train_op,
/external/tensorflow/tensorflow/contrib/eager/python/examples/rnn_ptb/
Drnn_ptb_graph_test.py48 train_op = optimizer.apply_gradients(grads)
52 sess.run(train_op, feed_dict={inputs_ph: inputs, labels_ph: labels})
54 [train_op, loss], feed_dict={
134 train_op = optimizer.apply_gradients(grads)
139 sess.run(train_op)
143 sess.run(train_op)
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder_test.cc181 CollectionDef train_op; in TEST_F() local
182 train_op.mutable_node_list()->add_value("y"); in TEST_F()
183 (*meta_graph.mutable_collection_def())["train_op"] = train_op; in TEST_F()
201 CollectionDef train_op; in TEST_F() local
202 train_op.mutable_node_list()->add_value("y"); in TEST_F()
203 (*meta_graph.mutable_collection_def())["train_op"] = train_op; in TEST_F()
258 CollectionDef train_op; in TEST_F() local
259 train_op.mutable_node_list()->add_value("y"); in TEST_F()
260 (*meta_graph.mutable_collection_def())["train_op"] = train_op; in TEST_F()
/external/tensorflow/tensorflow/contrib/eager/python/examples/resnet50/
Dresnet50_graph_test.py87 train_op = optimizer.minimize(loss)
99 sess.run([train_op, tf.contrib.summary.all_summary_ops()],
153 train_op = optimizer.minimize(loss)
160 sess.run(train_op)
163 sess.run(train_op)
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py114 train_op = optimizer.minimize(
121 self.evaluate(train_op)
218 train_op = optimizer.minimize(model(input_value))
223 self.evaluate(train_op)
309 train_op = optimizer.minimize(
324 session.run(train_op)
450 train_op = optimizer.minimize(root.var)
456 self.evaluate(train_op)
491 train_op = new_root.optimizer.minimize(new_root.var)
497 self.evaluate(train_op)
[all …]
/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_optimizer_v2_test.py69 train_op = optimizer.minimize(loss_fn, var_list=[var])
71 return train_op, optimizer
74 train_op, optimizer = distribution.extended.call_for_each_replica(
80 return distribution.group(train_op)
/external/tensorflow/tensorflow/contrib/tpu/
DREADME.md24 train_op = optimizer.minimize(loss, global_step=tf.train.get_global_step())
25 return tf.estimator.EstimatorSpec(mode=mode, loss=loss, train_op=train_op)

123456