Searched refs:_build_from_signature (Results 1 – 1 of 1) sorted by relevance
306 …layer._build_from_signature(query_shape, value_shape, key_shape) # pylint: disable=protected-acce…309 def _build_from_signature(self, query, value, key=None): member in MultiHeadAttention484 self._build_from_signature(query=query, value=value, key=key)