Lines Matching full:prepare

75     # This test checks whether performing quantization prepare before sparse prepare
87         tq.prepare(mod, inplace=True)
88         sparsifier.prepare(mod, config=sparse_config)
102     # This test checks whether performing sparsity prepare before quantization prepare
104     # the post sparse prepare module names (adding parametrizations changes the module class names)
116         sparsifier.prepare(mod, config=sparse_config)
117         tq.prepare(mod, inplace=True)
120         # that none were lost during prepare
147         sparsifier.prepare(mod, config=sparse_config)
148         tq.prepare(mod, inplace=True)
151         # that none were lost during prepare
175     # This tests whether performing sparse prepare before fusion causes any issues. The
186         sparsifier.prepare(mod, config=sparse_config)
189         tq.prepare(mod, inplace=True)
192         # that none were lost during prepare or fusion
205     # This tests whether performing fusion before sparse prepare causes and issues. The
228         sparsifier.prepare(mod, config=sparse_config)
230         tq.prepare(mod, inplace=True)
233         # that none were lost during prepare
257     # This tests whether performing sparse prepare before qat prepare causes issues.
269         sparsifier.prepare(mod, config=sparse_config)
287     # This tests whether performing qat prepare before sparse prepare causes issues.
304         sparsifier.prepare(mod, config=sparse_config)
307         # that none were lost during qat prepare
343         This test checks that the ordering of prepare_fx -> sparse prepare -> convert_fx
345         having to call squash mask between sparse prepare and convert_fx. This also tests the
375         sparsifier.prepare(mod, config=sparse_config)
378         # that none were lost during prepare
408         This checks that the ordering: prepare_fx -> sparse prepare -> convert_to_reference_fx
439         sparsifier.prepare(mod, config=sparse_config)
442         # that none were lost during prepare
475         This test checks that the ordering of sparse prepare -> prepare_fx -> convert_fx
484         sparsifier.prepare(mod, config=sparse_config)
496         # that none were lost during prepare
526         This test checks that the ordering of sparse prepare -> prepare_qat_fx -> convert_fx
535         sparsifier.prepare(mod, config=sparse_config)
547         # that none were lost during prepare
580         This checks that the ordering: sparse prepare -> prepare_fx -> convert_to_reference_fx
589         sparsifier.prepare(mod, config=sparse_config)
601         # that none were lost during prepare