Home
last modified time | relevance | path

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

/external/pytorch/torch/_inductor/fx_passes/
Dpad_mm.py755 def bmm_pattern(mat1: Tensor, mat2: Tensor) -> Tensor: function
844 typing.cast(SearchFn, bmm_pattern),
Dquantization.py2471 bmm_pattern = _generate_qlinear_weight_prepack_patterns(
2479 bmm_pattern,