Searched refs:_test_multihead_attn_invalid_shape_impl (Results 1 – 1 of 1) sorted by relevance
509 def _test_multihead_attn_invalid_shape_impl(self, mha): member in TestMultiheadAttentionNN600 self._test_multihead_attn_invalid_shape_impl(mha)604 self._test_multihead_attn_invalid_shape_impl(mha.eval())