Home
last modified time | relevance | path

Searched refs:_build_attention_equation (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dmulti_head_attention.py47 def _build_attention_equation(rank, attn_axes): function
411 _build_attention_equation(rank, attn_axes=self._attention_axes))