Searched refs:TupleSimplifier (Results 1 – 9 of 9) sorted by relevance
28 class TupleSimplifier : public HloModulePass {30 TupleSimplifier() : TupleSimplifier(/*exclude_entry_computation=*/false) {} in TupleSimplifier() function31 explicit TupleSimplifier(bool exclude_entry_computation);32 ~TupleSimplifier() override {} in ~TupleSimplifier()
33 TupleSimplifier::TupleSimplifier(bool exclude_entry_computation) : in TupleSimplifier() function in xla::TupleSimplifier36 StatusOr<bool> TupleSimplifier::Run(HloModule* module) { in Run()
596 EXPECT_TRUE(TupleSimplifier().Run(m.get()).ok()); in TEST_F()639 EXPECT_TRUE(TupleSimplifier().Run(m.get()).ok()); in TEST_F()706 EXPECT_TRUE(TupleSimplifier().Run(m.get()).ok()); in TEST_F()
40 TupleSimplifier simplifier; in Run()46 TupleSimplifier simplifier(exclude_entry); in Run()
1176 TupleSimplifier tuple_simplifier; in Run()1281 TupleSimplifier tuple_simplifier; in AddCopiesForBufferAssignment()
809 TupleSimplifier tuple_simplifier; in Run()
2118 TupleSimplifier tuple_simplifier; in ClearPreviousPassSideEffects()
235 pass.AddPass<TupleSimplifier>(); in OptimizeHloModule()281 pipeline.AddPass<TupleSimplifier>(); in OptimizeHloModule()348 pipeline.AddPass<TupleSimplifier>(); in OptimizeHloModule()
303 pass.AddPass<TupleSimplifier>(); in RunHloPassesThroughLayoutAssn()