| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | template.py | 343 trainable_variables = ops.get_collection_ref(
348 if trainable_at_start != len(trainable_variables):
352 (trainable_variables[trainable_at_start:],))
441 def trainable_variables(self): member in Template
454 trainable_variables = set(self.trainable_variables)
455 return [x for x in global_variables if x not in trainable_variables]
483 return self.trainable_variables
555 def trainable_variables(self): member in _EagerTemplateVariableStore
557 self._eager_variable_store.trainable_variables())
626 trainable_at_start = self._template_store.trainable_variables()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
| H A D | hvp_test.py | 30 model.trainable_variables, vector) as acc:
35 grads = grad_tape.gradient(loss, model.trainable_variables)
41 grad_tape.watch(model.trainable_variables)
43 model.trainable_variables, vector) as acc:
47 return grad_tape.gradient(acc.jvp(loss), model.trainable_variables)
55 variables = model.trainable_variables
68 grads = inner_tape.gradient(loss, model.trainable_variables)
70 grads, model.trainable_variables, output_gradients=vector)
91 vector = [tf.ones_like(v) for v in model.trainable_variables]
98 for hvp_component, variable in zip(hvp, model.trainable_variables):
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | template_test.py | 407 self.assertEqual(1, len(nested1.trainable_variables))
408 self.assertEqual(1, len(nested2.trainable_variables))
409 self.assertIs(nested1.trainable_variables[0], v1)
410 self.assertIs(nested2.trainable_variables[0], v2)
428 for v, w in zip(tmpl1.trainable_variables, [v1, v2]):
437 for v, w in zip(tmpl2.trainable_variables, [v5, v6]):
477 self.assertIs(nested1.trainable_variables[0], v1[0])
478 self.assertIs(nested2.trainable_variables[0], v2[0])
710 self.assertEqual([], list(ta.trainable_variables))
711 self.assertEqual([], list(tb.trainable_variables))
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ |
| H A D | gradients_test.py | 143 weights = variables.trainable_variables()
170 return gradient_ops.gradients(activation, variables.trainable_variables())
176 loop_fn_dtypes = [x.dtype for x in variables.trainable_variables()]
194 return gradient_ops.gradients(output, variables.trainable_variables())
205 loop_fn_dtypes = [x.dtype for x in variables.trainable_variables()]
290 return gradient_ops.gradients(loss, variables.trainable_variables())
294 loop_fn, [dtypes.float32] * len(variables.trainable_variables()),
317 logits, variables.trainable_variables(), use_pfor=use_pfor)
324 [dtypes.float32] * len(variables.trainable_variables()), batch_size)
333 jacobians = gradients.jacobian(output, variables.trainable_variables())
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | adadelta_test.py | 76 self.assertNotIn(slot[0], variables.trainable_variables())
80 self.assertNotIn(slot_update[0], variables.trainable_variables())
84 self.assertNotIn(slot[1], variables.trainable_variables())
88 self.assertNotIn(slot_update[1], variables.trainable_variables())
|
| H A D | momentum_test.py | 54 self.assertFalse(slot0 in variables.trainable_variables())
57 self.assertFalse(slot1 in variables.trainable_variables())
140 self.assertFalse(slot0 in variables.trainable_variables())
143 self.assertFalse(slot1 in variables.trainable_variables())
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | rnn_test.py | 371 trainable_variables = ops_lib.get_collection(
374 trainable_variables)
390 trainable_variables = ops_lib.get_collection(
393 trainable_variables)
504 trainable_variables = ops_lib.get_collection(
507 trainable_variables)
572 trainable_variables = ops_lib.get_collection(
575 trainable_variables)
638 trainable_variables = ops_lib.get_collection(
641 trainable_variables)
|
| H A D | rnn_cell_test.py | 1183 trainable_variables = ops.get_collection(
1185 assert len(trainable_variables) > 1, (
1186 "Count of trainable variables: %d" % len(trainable_variables))
1189 gradients_impl.gradients(y, trainable_variables)
1254 trainable_variables = ops.get_collection(
1256 assert len(trainable_variables) > 1, (
1257 "Count of trainable variables: %d" % len(trainable_variables))
1259 gradients_impl.gradients(y, trainable_variables)
2183 variables = variables_lib.trainable_variables()
2587 ], [v.name for v in cell.trainable_variables])
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | adadelta_test.py | 86 self.assertFalse(slot[0] in variables.trainable_variables())
90 self.assertFalse(slot_update[0] in variables.trainable_variables())
94 self.assertFalse(slot[1] in variables.trainable_variables())
98 self.assertFalse(slot_update[1] in variables.trainable_variables())
|
| H A D | momentum_test.py | 79 self.assertFalse(slot0 in variables.trainable_variables())
80 self.assertFalse(slot1 in variables.trainable_variables())
301 self.assertFalse(slot0 in variables.trainable_variables())
304 self.assertFalse(slot1 in variables.trainable_variables())
|
| H A D | moving_averages.py | 531 var_list = variables.trainable_variables()
672 moving_avg_variables = variables.trainable_variables()
|
| H A D | moving_averages_test.py | 191 self.assertNotIn(avg0, variables.trainable_variables())
192 self.assertNotIn(avg1, variables.trainable_variables())
193 self.assertNotIn(avg2, variables.trainable_variables())
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/trackable/ |
| H A D | data_structures.py | 246 trainable_variables = []
249 trainable_variables += obj.trainable_variables
253 return trainable_variables + trainable_extra_variables
270 trainable_variables = []
273 trainable_variables += obj.trainable_variables
276 trainable_variables + trainable_extra_variables +
289 def trainable_variables(self): member in TrackableDataStructure
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/module/ |
| H A D | module_test.py | 234 self.assertEqual(m.trainable_variables,
236 self.assertEqual(m.child.trainable_variables,
238 self.assertEqual(m.child.child.trainable_variables, (m.child.child.w,))
242 self.assertEqual(len(m.trainable_variables), 0)
243 self.assertEqual(len(m.child.trainable_variables), 0)
244 self.assertEqual(len(m.child.child.trainable_variables), 0)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training.py | 801 self.optimizer.minimize(loss, self.trainable_variables, tape=tape)
2044 trainable_variables = []
2046 trainable_variables += trackable_obj.trainable_variables
2047 trainable_variables += self._trainable_weights
2048 return self._dedup_weights(trainable_variables)
2059 trainable_variables = []
2061 trainable_variables += trackable_obj.trainable_variables
2064 trainable_variables + self._trainable_weights +
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| H A D | variable_scope_test.py | 199 self.assertTrue(v in store.trainable_variables())
200 self.assertFalse(w in store.trainable_variables())
213 self.assertTrue(new_v in new_store.trainable_variables())
214 self.assertFalse(new_w in new_store.trainable_variables())
1252 [v.name for v in scope.trainable_variables()],
1259 self.assertEqual([v.name for v in scope.trainable_variables()], [
1268 self.assertEqual([v.name for v in scope.trainable_variables()], [
1292 [v.name for v in scope.trainable_variables()],
1300 self.assertEqual([v.name for v in scope.trainable_variables()], [
1310 self.assertEqual([v.name for v in scope.trainable_variables()], [
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | single_loss_example.py | 69 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
112 return optimizer.minimize(loss_fn, lambda: layer.trainable_variables)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/ |
| H A D | functional_ops_test.py | 115 self.assertEqual(len(variables.trainable_variables()), 1)
116 self.assertEqual(variables.trainable_variables()[0].name,
124 self.assertEqual(len(variables.trainable_variables()), 1)
168 self.assertEqual(len(variables.trainable_variables()), 1)
169 self.assertEqual(variables.trainable_variables()[0].name,
177 self.assertEqual(len(variables.trainable_variables()), 1)
273 self.assertEqual(len(variables.trainable_variables()), 1)
274 self.assertEqual(variables.trainable_variables()[0].name,
283 self.assertEqual(len(variables.trainable_variables()), 1)
|
| H A D | map_fn_test.py | 133 self.assertEqual(len(variables.trainable_variables()), 1)
134 self.assertEqual(variables.trainable_variables()[0].name,
142 self.assertEqual(len(variables.trainable_variables()), 1)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| H A D | base.py | 486 trainable_variables = tf_variables.trainable_variables()
488 variable not in trainable_variables):
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| H A D | mnist_train.py | 148 grads = tape.gradient(loss_value, model.trainable_variables)
151 optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/integration_test/ |
| H A D | tpu_memory_test.py | 151 grads = tape.gradient(loss, model.trainable_variables)
152 model.optimizer.apply_gradients(zip(grads, model.trainable_variables))
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.-module.pbtxt | 24 name: "trainable_variables"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.__internal__.tracking.-trackable-data-structure.pbtxt | 27 name: "trainable_variables"
|
| H A D | tensorflow.-module.pbtxt | 24 name: "trainable_variables"
|