Searched refs:key_dim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention_test.py | 44 key_dim=64,
56 num_heads=12, key_dim=64)
65 num_heads=12, key_dim=64)
75 num_heads=12, key_dim=64)
87 num_heads=2, key_dim=2, use_bias=use_bias)
138 key_dim=64,
148 num_heads=2, key_dim=2)
201 num_heads=2, key_dim=2, attention_axes=attention_axes)
230 num_heads=2, key_dim=2, dropout=0.5)
264 test_layer = SubclassAttention(num_heads=12, key_dim=64)
[all …]
|
| D | multi_head_attention.py | 219 key_dim, argument
235 self._key_dim = key_dim
236 self._value_dim = value_dim if value_dim else key_dim
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-multi-head-attention.pbtxt | 132 …argspec: "args=[\'self\', \'num_heads\', \'key_dim\', \'value_dim\', \'dropout\', \'use_bias\', \'…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-multi-head-attention.pbtxt | 132 …argspec: "args=[\'self\', \'num_heads\', \'key_dim\', \'value_dim\', \'dropout\', \'use_bias\', \'…
|