Searched refs:ShapeOptimizer (Results 1 – 5 of 5) 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()
47 ShapeOptimizer optimizer; in TEST_F()86 ShapeOptimizer optimizer; in TEST_F()
31 Status ShapeOptimizer::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()135 void ShapeOptimizer::Feedback(Cluster* /*cluster*/, in Feedback()
128 MK_OPT("shape", new ShapeOptimizer()); in MakeNewOptimizer()179 optimizers->push_back(MakeUnique<ShapeOptimizer>()); in InitializeOptimizers()
109 enabled_optimizers_["shape"] = MakeUnique<ShapeOptimizer>(); in Init()