Searched refs:NormalizeArgs (Results 1 – 2 of 2) sorted by relevance
19 class NormalizeArgs(Transformer): class
21 from torch.fx.experimental.normalize import NormalizeArgs, NormalizeOperators943 traced = NormalizeArgs(traced).transform()1037 traced = NormalizeArgs(symbolic_trace(test_instance)).transform()1063 traced = NormalizeArgs(traced).transform()1080 traced = NormalizeArgs(traced).transform()