Searched refs:_softmax (Results 1 – 2 of 2) sorted by relevance
414 self._softmax = advanced_activations.Softmax(axis=norm_axes)427 return self._softmax(attention_scores, attention_mask)
102 def _softmax(self, x): member in SoftmaxTest113 y_np = self._softmax(x_np)