| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/ |
| D | sdca_ops_test.py | 265 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/ |
| D | basic_session_run_hooks_test.py | 250 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/ |
| D | learning_test.py | 254 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/ |
| D | training_test.py | 101 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 …]
|
| D | training.py | 466 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/ |
| D | datasets_test.py | 53 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 …]
|
| D | cluster_test.py | 39 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 …]
|
| D | item_test.py | 55 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)
|
| D | memory_optimizer_test.py | 48 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 …]
|
| D | model_analyzer_test.py | 39 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)
|
| D | cost_analyzer_test.py | 48 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/ |
| D | graph_actions_test.py | 473 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 …]
|
| D | monitors_test.py | 624 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/ |
| D | model_fn.py | 90 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,
|
| D | estimator_input_test.py | 114 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/ |
| D | constrained_optimizer.py | 138 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/ |
| D | builder_impl.py | 232 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/ |
| D | gbdt_batch_test.py | 259 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/ |
| D | learning.py | 61 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/ |
| D | rnn_ptb_graph_test.py | 48 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/ |
| D | grappler_item_builder_test.cc | 181 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/ |
| D | resnet50_graph_test.py | 87 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/ |
| D | checkpointable_utils_test.py | 114 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/ |
| D | keras_optimizer_v2_test.py | 69 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/python/training/tracking/ |
| D | util_with_v1_optimizers_test.py | 96 train_op = optimizer.minimize(
103 self.evaluate(train_op)
200 train_op = optimizer.minimize(model(input_value))
205 self.evaluate(train_op)
336 train_op = strategy.extended.call_for_each_replica(
343 session.run(train_op)
362 train_op = optimizer.minimize(
380 session.run(train_op)
513 train_op = optimizer.minimize(root.var)
519 self.evaluate(train_op)
[all …]
|