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()
|