| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | sharded_variable_test.py | 50 self.assertLen(layer.trainable_weights, 2)
51 self.assertEqual(layer.trainable_weights[0], [0])
52 self.assertEqual(layer.trainable_weights[1], [1])
57 layer.trainable_weights + layer.non_trainable_weights)
58 self.assertAllEqual(layer.trainable_weights, layer.trainable_variables)
94 self.assertLen(layer.trainable_weights, 2)
95 self.assertEqual(layer.trainable_weights[0], [0])
96 self.assertEqual(layer.trainable_weights[1], [1])
101 layer.trainable_weights + layer.non_trainable_weights)
102 self.assertAllEqual(layer.trainable_weights, layer.trainable_variables)
|
| /external/tensorflow/tensorflow/python/keras/tests/ |
| D | custom_training_loop_test.py | 81 gradients = tape.gradient(loss, model.trainable_weights)
82 optimizer.apply_gradients(zip(gradients, model.trainable_weights))
104 gradients = tape.gradient(loss, model.trainable_weights)
105 optimizer.apply_gradients(zip(gradients, model.trainable_weights))
129 gradients = tape.gradient(loss, model.trainable_weights)
130 optimizer.apply_gradients(zip(gradients, model.trainable_weights))
|
| D | model_subclassing_compiled_test.py | 131 self.assertEqual(len(model.trainable_weights), 8)
324 self.assertEqual(len(model.trainable_weights),
325 6 + len(model.test_net.trainable_weights))
348 self.assertEqual(len(model.trainable_weights),
349 6 + len(model.test_net.trainable_weights))
372 self.assertEqual(len(model.trainable_weights), 12)
406 self.assertEqual(len(model.trainable_weights), 6)
|
| D | tracking_test.py | 113 self.assertTrue(any(model.layer_list[0].trainable_weights[0]
114 is t for t in model.trainable_weights))
119 self.assertIn(model.v, model.trainable_weights)
122 self.assertIn(model.v, model2.trainable_weights)
141 self.assertIn(layer.kernel, model2.trainable_weights)
293 self.assertEqual([], root.wrapper.trainable_weights)
300 self.assertEqual([], root.wrapper.trainable_weights)
453 self.assertIn(id(model.layer_list[0].trainable_weights[0]),
454 [id(obj) for obj in model.trainable_weights])
485 self.assertIn(layer.kernel, model2.trainable_weights)
|
| D | add_loss_correctness_test.py | 54 gradients = tape.gradient(loss, model.trainable_weights)
55 optimizer.apply_gradients(zip(gradients, model.trainable_weights))
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_eager_v1.py | 268 trainable_weights = model.trainable_weights
269 if trainable_weights:
275 grads = tape.gradient(scaled_total_loss, trainable_weights)
279 model.optimizer.apply_gradients(zip(grads, trainable_weights))
|
| D | training_test.py | 777 self.assertLen(model.trainable_weights, 4)
2165 self.assertListEqual(model.trainable_weights, [])
2171 self.assertListEqual(model.trainable_weights, layer.trainable_weights)
2173 self.assertListEqual(model.trainable_weights, [])
2179 self.assertListEqual(model.trainable_weights, [])
2186 self.assertListEqual(model.trainable_weights, layer.trainable_weights)
2188 self.assertListEqual(model.trainable_weights, [])
2196 self.assertListEqual(model.trainable_weights, [])
2202 self.assertListEqual(model.trainable_weights, [])
2212 self.assertListEqual(outer_model.trainable_weights,
[all …]
|
| D | sequential_test.py | 230 self.assertEqual(len(model.trainable_weights), 4)
232 self.assertEqual(len(model.trainable_weights), 2)
234 self.assertEqual(len(model.trainable_weights), 4)
|
| D | functional_test.py | 279 self._assertAllIs(network.trainable_weights, dense.trainable_weights)
296 self.assertEqual(network.trainable_weights, [])
298 dense.trainable_weights + dense.non_trainable_weights)
307 self._assertAllIs(model.trainable_weights, weights)
311 self.assertListEqual(model.trainable_weights, [])
315 self._assertAllIs(model.trainable_weights, weights)
319 self.assertListEqual(model.trainable_weights, [])
327 self._assertAllIs(model.trainable_weights, weights)
331 self.assertListEqual(model.trainable_weights, [])
335 self._assertAllIs(model.trainable_weights, weights)
[all …]
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | hdf5_format.py | 298 trainable_weights = weights[:len(layer.trainable_weights)]
299 non_trainable_weights = weights[len(layer.trainable_weights):]
305 num_trainable_weights = len(sublayer.trainable_weights)
310 weights=(trainable_weights[:num_trainable_weights] +
317 trainable_weights = trainable_weights[num_trainable_weights:]
898 weights = layer.trainable_weights + layer.non_trainable_weights
|
| /external/tensorflow/tensorflow/python/keras/premade/ |
| D | wide_deep.py | 167 params=self.linear_model.trainable_weights, # pylint: disable=protected-access
171 params=self.dnn_model.trainable_weights, # pylint: disable=protected-access
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core_test.py | 298 self.assertLen(layer.trainable_weights, 0)
300 self.assertLen(layer.trainable_weights, 1)
309 self.assertLen(model.trainable_weights, 1)
310 self.assertAllClose(keras.backend.get_value(model.trainable_weights[0]), 2.)
|
| D | cudnn_recurrent_test.py | 259 self.assertEqual(len(layer.trainable_weights), 3)
264 self.assertEqual(len(layer.trainable_weights), 0)
267 self.assertEqual(len(layer.trainable_weights), 3)
|
| D | normalization_test.py | 83 self.assertEqual(len(layer.trainable_weights), 0)
88 self.assertEqual(len(layer.trainable_weights), 2)
581 self.assertEqual(len(layer.trainable_weights), 0)
586 self.assertEqual(len(layer.trainable_weights), 2)
|
| D | wrappers_test.py | 237 self.assertEqual(len(layer.trainable_weights), 2)
239 assert not layer.trainable_weights
241 assert len(layer.trainable_weights) == 2
841 assert len(layer.trainable_weights) == 6
843 assert not layer.trainable_weights
845 assert len(layer.trainable_weights) == 6
|
| /external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| D | rnn_cell_impl.py | 1297 def trainable_weights(self): member in MultiRNNCell
1303 weights += cell.trainable_weights
1313 trainable_weights = []
1316 trainable_weights += cell.trainable_weights
1317 return trainable_weights + weights
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 84 grads = tape.gradient(loss, model.trainable_weights)
85 optimizer.apply_gradients(zip(grads, model.trainable_weights))
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | data_structures.py | 219 def trainable_weights(self): member in TrackableDataStructure
262 return self.trainable_weights + self.non_trainable_weights
266 return self.trainable_weights
|
| D | data_structures_test.py | 84 self.assertEqual([v], l.trainable_weights)
94 self.assertEqual([v], l.trainable_weights)
529 self.assertEqual([v], l.trainable_weights)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | layer_utils.py | 269 trainable_count = count_params(model.trainable_weights)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-re-l-u.pbtxt | 111 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-masking.pbtxt | 111 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-thresholded-re-l-u.pbtxt | 111 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-cropping2-d.pbtxt | 111 name: "trainable_weights"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-zero-padding2-d.pbtxt | 111 name: "trainable_weights"
|