Searched refs:BaseDenseAttention (Results 1 – 6 of 6) sorted by relevance
38 class BaseDenseAttention(Layer): class82 super(BaseDenseAttention, self).__init__(**kwargs)220 base_config = super(BaseDenseAttention, self).get_config()225 class Attention(BaseDenseAttention):363 class AdditiveAttention(BaseDenseAttention):
47 actual, actual_scores = dense_attention.BaseDenseAttention()._apply_scores(63 actual, actual_scores = dense_attention.BaseDenseAttention()._apply_scores(81 actual, actual_scores = dense_attention.BaseDenseAttention()._apply_scores(103 actual, actual_scores = dense_attention.BaseDenseAttention()._apply_scores(130 actual, actual_scores = dense_attention.BaseDenseAttention()._apply_scores(151 actual, actual_scores = dense_attention.BaseDenseAttention(164 layer = dense_attention.BaseDenseAttention(causal=True)171 new_layer = dense_attention.BaseDenseAttention.from_config(config)
4 is_instance: "<class \'tensorflow.python.keras.layers.dense_attention.BaseDenseAttention\'>"