| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | network_test.py | 474 self.assertIsInstance(net.trainable_weights[0],
553 actual=net1.trainable_weights[0].name)
557 actual=net2.trainable_weights[0].name)
562 actual=net3.trainable_weights[0].name)
568 actual=outside_net_before.trainable_weights[0].name)
572 actual=outside_net_after.trainable_weights[0].name)
584 actual=net.trainable_weights[0].name)
615 actual=net.trainable_weights[0].name)
671 self.evaluate(net1.trainable_weights[0].assign([[1.]]))
672 self.evaluate(net2.trainable_weights[0].assign([[2.]]))
[all …]
|
| D | network.py | 463 def trainable_weights(self): member in Network
466 weights += layer.trainable_weights
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | layer_utils.py | 76 weights += layer.trainable_weights
106 trainable_weights = []
108 trainable_weights += layer.trainable_weights
109 return (trainable_weights + trainable_extra_variables
|
| D | data_structures.py | 176 def trainable_weights(self): member in TrackableDataStructure
191 return self.trainable_weights + self.non_trainable_weights
195 return self.trainable_weights
|
| D | data_structures_test.py | 197 self.assertEqual([v], l.trainable_weights)
207 self.assertEqual([v], l.trainable_weights)
472 self.assertEqual([], root.wrapper.trainable_weights)
479 self.assertEqual([], root.wrapper.trainable_weights)
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | layer_utils.py | 234 trainable_count = count_params(model.trainable_weights)
261 weights += layer.trainable_weights
291 trainable_weights = []
293 trainable_weights += layer.trainable_weights
294 return (trainable_weights + trainable_extra_variables
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_eager.py | 221 if not model.trainable_weights:
226 grads = tape.gradient(total_loss, model.trainable_weights)
228 model.trainable_weights))
|
| D | training_test.py | 1586 self.assertListEqual(model.trainable_weights, [])
1592 self.assertListEqual(model.trainable_weights, layer.trainable_weights)
1594 self.assertListEqual(model.trainable_weights, [])
1600 self.assertListEqual(model.trainable_weights, [])
1607 self.assertListEqual(model.trainable_weights, layer.trainable_weights)
1609 self.assertListEqual(model.trainable_weights, [])
1618 self.assertListEqual(model.trainable_weights, [])
1624 self.assertListEqual(model.trainable_weights, [])
1635 self.assertListEqual(outer_model.trainable_weights,
1636 inner_model.trainable_weights)
[all …]
|
| D | topology_test.py | 290 self.assertEqual(network.trainable_weights, dense.trainable_weights)
306 self.assertEqual(network.trainable_weights, [])
308 dense.trainable_weights + dense.non_trainable_weights)
317 self.assertListEqual(model.trainable_weights, weights)
321 self.assertListEqual(model.trainable_weights, [])
325 self.assertListEqual(model.trainable_weights, weights)
329 self.assertListEqual(model.trainable_weights, [])
337 self.assertListEqual(model.trainable_weights, weights)
341 self.assertListEqual(model.trainable_weights, [])
345 self.assertListEqual(model.trainable_weights, weights)
[all …]
|
| D | sequential_test.py | 238 self.assertEqual(len(model.trainable_weights), 4)
240 self.assertEqual(len(model.trainable_weights), 2)
242 self.assertEqual(len(model.trainable_weights), 4)
|
| D | base_layer_test.py | 492 self.assertEqual(len(layer.trainable_weights), 5)
497 self.assertEqual(len(layer.trainable_weights), 3)
502 self.assertEqual(len(layer.trainable_weights), 0)
565 self.assertEqual([], l.trainable_weights)
|
| D | training.py | 355 self._collected_trainable_weights = self.trainable_weights
451 trainable_weights = self.trainable_weights
452 self._collected_trainable_weights = trainable_weights
2066 if len(self.trainable_weights) != len(self._collected_trainable_weights):
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core_test.py | 204 self.assertLen(layer.trainable_weights, 1)
205 self.assertEqual(layer.trainable_weights[0].name, 'lambda/multiplier:0')
227 self.assertLen(model.trainable_weights, 1)
228 self.assertAllClose(keras.backend.get_value(model.trainable_weights[0]), 2.)
|
| D | wrappers_test.py | 187 self.assertEqual(len(layer.trainable_weights), 2)
190 assert not layer.trainable_weights
193 assert len(layer.trainable_weights) == 2
537 assert len(layer.trainable_weights) == 6
539 assert not layer.trainable_weights
541 assert len(layer.trainable_weights) == 6
|
| D | normalization_test.py | 68 self.assertEqual(len(layer.trainable_weights), 0)
73 self.assertEqual(len(layer.trainable_weights), 2)
435 self.assertEqual(len(layer.trainable_weights), 0)
440 self.assertEqual(len(layer.trainable_weights), 2)
|
| D | cudnn_recurrent_test.py | 253 self.assertEqual(len(layer.trainable_weights), 3)
258 self.assertEqual(len(layer.trainable_weights), 0)
261 self.assertEqual(len(layer.trainable_weights), 3)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | model_subclassing_test.py | 533 self.assertEqual(1, len(model.trainable_weights))
549 self.assertEqual(1, len(model.trainable_weights))
675 self.assertEqual(len(model.trainable_weights), 8)
864 self.assertEqual(len(model.trainable_weights),
865 6 + len(model.test_net.trainable_weights))
888 self.assertEqual(len(model.trainable_weights),
889 6 + len(model.test_net.trainable_weights))
911 self.assertEqual(len(model.trainable_weights), 12)
945 self.assertEqual(len(model.trainable_weights), 6)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | rnn_cell_impl.py | 1659 def trainable_weights(self): member in MultiRNNCell
1665 weights += cell.trainable_weights
1675 trainable_weights = []
1678 trainable_weights += cell.trainable_weights
1679 return trainable_weights + weights
|
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| D | containers_test.py | 106 self.assertEqual(2, len(tracker.trainable_weights))
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | hdf5_format.py | 347 num_weights = len(sublayer.trainable_weights)
359 if l not in sublayer.trainable_weights])
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-max-pooling3-d.pbtxt | 73 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-zero-padding2-d.pbtxt | 72 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-activation.pbtxt | 72 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-gaussian-noise.pbtxt | 72 name: "trainable_weights"
|
| D | tensorflow.keras.layers.-max-pooling2-d.pbtxt | 73 name: "trainable_weights"
|