Searched refs:DisableReduction (Results 1 – 2 of 2) sorted by relevance
1100 node_schedule.append(DisableReduction)1111 node_schedule[-1], (EnableReduction, DisableReduction)1238 lambda n: n not in (EnableReduction, DisableReduction)1261 lambda n: n not in (EnableReduction, DisableReduction)1285 if node in (DisableReduction, EnableReduction):1323 if node in (EnableReduction, DisableReduction):1375 if node not in (EnableReduction, DisableReduction):1414 return itertools.takewhile(lambda n: n is not DisableReduction, nodes)1423 if node is DisableReduction:1440 if node is DisableReduction:[all …]
1858 from torch._inductor.codegen.simd import DisableReduction, EnableReduction1866 elif node is DisableReduction: