Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Djoint_graph.py643 def mul_softmax_pattern(match: Match, *, inp, other, dim, keepdim, dtype=None): function
667 )(mul_softmax_pattern)