Searched refs:use_scale (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | dense_attention.py | 325 def __init__(self, use_scale=False, **kwargs): argument
327 self.use_scale = use_scale
331 if self.use_scale:
357 config = {'use_scale': self.use_scale}
465 def __init__(self, use_scale=True, **kwargs): argument
467 self.use_scale = use_scale
474 if self.use_scale:
499 if self.use_scale:
507 config = {'use_scale': self.use_scale}
|
| D | dense_attention_test.py | 235 attention_layer = dense_attention.Attention(use_scale=True)
394 attention_layer = dense_attention.Attention(use_scale=True)
401 attention_layer = dense_attention.Attention(use_scale=True)
511 def test_serialization(self, use_scale): argument
513 layer = dense_attention.Attention(use_scale=use_scale)
517 self.assertEqual(new_layer.use_scale, use_scale)
521 self.assertEqual(new_layer.use_scale, use_scale)
611 attention_layer = dense_attention.AdditiveAttention(use_scale=False)
754 layer = dense_attention.AdditiveAttention(use_scale=True)
758 self.assertEqual(new_layer.use_scale, True)
[all …]
|
| /external/skqp/samplecode/ |
| D | SampleIdentityScale.cpp | 55 bool use_scale = (time.fSecond % 2 == 1); in onDrawContent() local
59 if (use_scale) { in onDrawContent()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.keras.layers.-attention.pbtxt | 133 argspec: "args=[\'self\', \'use_scale\'], varargs=None, keywords=kwargs, defaults=[\'False\'], "
|
| D | tensorflow.keras.layers.-additive-attention.pbtxt | 133 argspec: "args=[\'self\', \'use_scale\'], varargs=None, keywords=kwargs, defaults=[\'True\'], "
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.keras.layers.-additive-attention.pbtxt | 133 argspec: "args=[\'self\', \'use_scale\'], varargs=None, keywords=kwargs, defaults=[\'True\'], "
|
| D | tensorflow.keras.layers.-attention.pbtxt | 133 argspec: "args=[\'self\', \'use_scale\'], varargs=None, keywords=kwargs, defaults=[\'False\'], "
|