Searched refs:ShapeOptimizer (Results 1 – 4 of 4) sorted by relevance
31 class ShapeOptimizer : public GraphOptimizer {33 ShapeOptimizer() {} in ShapeOptimizer() function34 explicit ShapeOptimizer(RewriterConfig::Toggle opt_level) {} in ShapeOptimizer() function36 ~ShapeOptimizer() override {} in ~ShapeOptimizer()
51 ShapeOptimizer optimizer; in TEST_F()113 ShapeOptimizer optimizer; in TEST_F()147 ShapeOptimizer optimizer; in TEST_F()
37 Status ShapeOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()195 void ShapeOptimizer::Feedback(Cluster* /*cluster*/, in Feedback()
194 MK_OPT("shape", new ShapeOptimizer()); in MakeNewOptimizer()261 optimizers->push_back(MakeUnique<ShapeOptimizer>()); in InitializeOptimizers()