Searched refs:ModelPruner (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | model_pruner.h | 27 class ModelPruner : public GraphOptimizer { 29 ModelPruner() {} in ModelPruner() function 30 ~ModelPruner() override {} in ~ModelPruner()
|
D | model_pruner_test.cc | 46 ModelPruner pruner; in TEST_F() 68 ModelPruner pruner; in TEST_F() 111 ModelPruner pruner; in TEST_F() 156 ModelPruner pruner; in TEST_F() 204 ModelPruner pruner; in TEST_F() 245 ModelPruner pruner; in TEST_F() 296 ModelPruner pruner; in TEST_F() 331 ModelPruner pruner; in TEST_F() 422 ModelPruner pruner; in TEST_F() 468 ModelPruner pruner; in TEST_F() [all …]
|
D | arithmetic_optimizer_test_utils.h | 38 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, *item, output)); in OptimizeAndPrune() 53 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, *item, output)); in DedupAndOptimizeTwiceAndPrune() 84 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, *item, output));
|
D | model_pruner.cc | 421 Status ModelPruner::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize() 532 void ModelPruner::Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
|
D | meta_optimizer.cc | 185 MK_OPT("pruning", new ModelPruner()); in MakeNewOptimizer() 236 optimizers->push_back(MakeUnique<ModelPruner>()); in InitializeOptimizers()
|
D | arithmetic_optimizer_test.cc | 1627 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, item, &output)); in TEST_F() 1659 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, item, &output)); in TEST_F()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | grappler_item_builder.cc | 82 ModelPruner pruner; in PruneGraph()
|