Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dmisc_patterns.py131 numpy_compat_normalization = NumpyCompatNormalization() variable
H A Dpre_grad.py29 from .misc_patterns import numpy_compat_normalization
234 numpy_compat_normalization(gm.graph)
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_split_cat_fx_passes.py6 from torch._inductor.fx_passes.misc_patterns import numpy_compat_normalization
1502 numpy_compat_normalization(fn_t.graph)