Searched refs:ModelPruner (Results 1 – 8 of 8) 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 | 41 ModelPruner pruner; in TEST_F()
72 ModelPruner pruner; in TEST_F()
115 ModelPruner pruner; in TEST_F()
164 ModelPruner pruner; in TEST_F()
223 ModelPruner pruner; in TEST_F()
267 ModelPruner pruner; in TEST_F()
321 ModelPruner pruner; in TEST_F()
355 ModelPruner pruner; in TEST_F()
451 ModelPruner pruner; in TEST_F()
489 ModelPruner pruner; in TEST_F()
|
| D | arithmetic_optimizer_test_utils.h | 37 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, *item, output)); in OptimizeAndPrune()
68 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, *item, output));
|
| D | model_pruner.cc | 423 Status ModelPruner::Optimize(Cluster* cluster, const GrapplerItem& item, in Optimize()
518 void ModelPruner::Feedback(Cluster* cluster, const GrapplerItem& item, in Feedback()
|
| D | meta_optimizer.cc | 125 MK_OPT("pruning", new ModelPruner()); in MakeNewOptimizer()
162 optimizers->push_back(MakeUnique<ModelPruner>()); in InitializeOptimizers()
|
| D | arithmetic_optimizer_test.cc | 1606 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, item, &output)); in TEST_F()
1636 TF_EXPECT_OK(ModelPruner().Optimize(nullptr, item, &output)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | meta_optimizer.cc | 106 enabled_optimizers_["pruning"] = MakeUnique<ModelPruner>(); in Init()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.cc | 158 ModelPruner pruner; in PruneGraph()
|