Searched refs:_partial_softmax_pattern (Results 1 – 1 of 1) sorted by relevance
589 def _partial_softmax_pattern(linear_func, reverse=False, to_dtype=False): function664 _partial_softmax_pattern(aten.mul.Tensor, reverse=reverse, to_dtype=to_dtype),691 _partial_softmax_pattern(aten.div.Tensor, to_dtype=to_dtype),