| /external/tensorflow/tensorflow/python/training/ |
| D | basic_session_run_hooks_test.py | 249 train_op = constant_op.constant(3)
257 mon_sess.run(train_op)
267 train_op = constant_op.constant(3)
273 mon_sess.run(train_op)
278 mon_sess.run(train_op)
281 mon_sess.run(train_op)
286 mon_sess.run(train_op)
314 train_op = constant_op.constant(3)
320 mon_sess.run(train_op)
327 train_op = constant_op.constant(3)
[all …]
|
| D | sync_replicas_optimizer_test.py | 65 train_op = [
82 train_ops.append(train_op)
89 def _run(self, train_op, sess): argument
90 sess.run(train_op)
|
| D | optimizer.py | 644 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
645 if apply_updates not in train_op:
646 train_op.append(apply_updates)
739 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
740 if apply_updates not in train_op:
741 train_op.append(apply_updates)
|
| /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)
154 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
155 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)
74 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
75 train_op.append(d)
131 train_op = optimizer.minimize(loss)
134 return (metagraph, init_op.name, train_op.name, loss.name)
236 train_op = graph.get_operation_by_name(train_op_name)
240 self.evaluate(train_op)
241 self.evaluate(train_op)
288 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/python/saved_model/ |
| D | builder_impl.py | 236 train_op=None, argument
281 _add_op_to_signature_def_map(signature_def_map, train_op,
309 train_op=None, argument
359 _add_op_to_signature_def_map(signature_def_map, train_op,
439 def _add_collections(self, assets_collection, main_op, train_op): argument
446 self._add_train_op(train_op)
494 def _add_train_op(self, train_op): argument
507 if train_op is not None:
508 if (not isinstance(train_op, ops.Tensor) and
509 not isinstance(train_op, ops.Operation)):
[all …]
|
| D | loader_impl.py | 216 train_op = _get_op_from_signature_def(
218 if train_op is None:
219 train_op = _get_op_from_collection(meta_graph_def, constants.TRAIN_OP_KEY)
220 return 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()
354 key: "train_op" in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_with_v1_optimizers_test.py | 52 train_op = optimizer.minimize(root.var)
58 self.evaluate(train_op)
94 train_op = new_root.optimizer.minimize(new_root.var)
100 self.evaluate(train_op)
150 train_op = optimizer.minimize(first_variable.read_value)
153 self.evaluate(train_op)
166 train_op = optimizer.minimize(second_variable.read_value)
168 self.evaluate(train_op)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | keras_optimizer_v2_test.py | 68 train_op = optimizer.minimize(loss_fn, var_list=[var])
70 return train_op, optimizer
73 train_op, optimizer = distribution.extended.call_for_each_replica(
79 return distribution.group(train_op)
|
| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| D | file_input_yielder.cc | 106 for (const auto& train_op : train_ops) { in NextItem() local
107 if (train_ops_found.find(train_op) != train_ops_found.end()) { in NextItem()
108 LOG(ERROR) << "Non existent train op specified: " << train_op; in NextItem()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_utils_test.py | 203 def __new__(cls, mode, loss, train_op, scaffold=None): argument
205 cls, mode=mode, loss=loss, train_op=train_op,
215 train_op=array_ops.identity(constant_op.constant(device_id)))
226 self.assertEqual(created_estimator_specs[device_id].train_op,
227 merged_estimator_spec.train_op.values[device_id])
|
| D | zero_batch_test.py | 69 train_op = optimizer.minimize(loss)
70 with ops.control_dependencies([train_op]):
73 train_op = distribution.extended.call_for_each_replica(
92 np_output, _, _ = sess.run([train_op] + bn.updates, {
|
| /external/tensorflow/tensorflow/python/keras/integration_test/ |
| D | legacy_rnn_test.py | 78 train_op = tf.train.GradientDescentOptimizer(0.001).minimize(loss)
82 [train_op, outputs, state], {inputs: x_train, predict: y_train})
111 train_op = tf.train.GradientDescentOptimizer(0.001).minimize(loss)
115 [train_op, outputs, state], {inputs: x_train, predict: y_train})
146 train_op = tf.train.GradientDescentOptimizer(0.001).minimize(loss)
150 [train_op, outputs, state], {inputs: x_train, predict: y_train})
189 train_op = tf.train.GradientDescentOptimizer(0.001).minimize(loss)
193 [train_op, outputs, state], {inputs: x_train, predict: y_train})
226 train_op = tf.train.GradientDescentOptimizer(0.001).minimize(loss)
232 [train_op, outputs, state], feed_dict)
|
| /external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| D | resnet50_graph_test.py | 115 train_op = optimizer.minimize(loss)
122 sess.run(train_op)
125 sess.run(train_op)
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | ram_file_system_test.py | 110 train_op = opt.minimize(
113 return EstimatorSpec(mode=mode, loss=loss, train_op=train_op)
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saved_model_experimental.py | 284 train_op = None
287 train_op = clone.train_function.updates_op
326 train_op=train_op)
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | tracking_util_with_v1_optimizers_test.py | 94 train_op = optimizer.minimize(
101 self.evaluate(train_op)
195 train_op = optimizer.minimize(model(input_value))
200 self.evaluate(train_op)
333 train_op = strategy.extended.call_for_each_replica(
340 session.run(train_op)
359 train_op = optimizer.minimize(
377 session.run(train_op)
580 train_op = optimizer.minimize(
585 self.evaluate(train_op)
|
| D | tracking_util_test.py | 120 train_op = control_flow_ops.group(
132 self.evaluate(train_op)
210 train_op = optimizer.apply_gradients(zip(gradients, variables))
214 self.evaluate(train_op)
309 train_op = optimizer.apply_gradients(zip(gradients, variables))
326 session.run(train_op)
476 train_op = optimizer.apply_gradients(zip(gradients, variables))
482 self.evaluate(train_op)
520 train_op = new_root.optimizer.apply_gradients(zip(gradients, variables))
530 self.evaluate(train_op)
[all …]
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | async_checkpoint_test.py | 75 train_op = optimizer.minimize(loss, training.get_global_step())
79 train_op=train_op,
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | quantization_mnist_test.py | 235 train_op = optimizer.minimize(loss, global_step=get_global_step())
236 return EstimatorSpec(mode, loss=loss, train_op=train_op)
|