Home
last modified time | relevance | path

Searched refs:ArithmeticOptimizer (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test_utils.h32 void OptimizeAndPrune(ArithmeticOptimizer* optimizer, GrapplerItem* item, in OptimizeAndPrune()
41 void OptimizeTwice(ArithmeticOptimizer* optimizer, GrapplerItem* item, in OptimizeTwice()
51 void OptimizeTwiceAndPrune(ArithmeticOptimizer* optimizer, GrapplerItem* item,
73 void DisableAllStages(ArithmeticOptimizer* optimizer) { in DisableAllStages()
74 ArithmeticOptimizer::ArithmeticOptimizerOptions options; in DisableAllStages()
102 void DisableAddToAddNCombining(ArithmeticOptimizer* optimizer) { in DisableAddToAddNCombining()
106 void EnableOnlyAddToAddNCombining(ArithmeticOptimizer* optimizer) { in EnableOnlyAddToAddNCombining()
111 void EnableOnlyFoldConjugateIntoTranspose(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldConjugateIntoTranspose()
116 void EnableOnlyFoldMultipleIntoConv(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldMultipleIntoConv()
121 void EnableOnlyFoldTransposeIntoMatMul(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldTransposeIntoMatMul()
[all …]
Darithmetic_optimizer.h33 class ArithmeticOptimizer : public GraphOptimizer {
35 ArithmeticOptimizer() in ArithmeticOptimizer() function
39 explicit ArithmeticOptimizer(RewriterConfig::Toggle opt_level) in ArithmeticOptimizer() function
43 ~ArithmeticOptimizer() override {} in ~ArithmeticOptimizer()
Darithmetic_optimizer_test.cc96 ArithmeticOptimizer optimizer; in TEST_F()
115 ArithmeticOptimizer optimizer; in TEST_F()
154 ArithmeticOptimizer optimizer; in TEST_F()
187 ArithmeticOptimizer optimizer; in TEST_F()
227 ArithmeticOptimizer optimizer; in TEST_F()
264 ArithmeticOptimizer optimizer; in TEST_F()
297 ArithmeticOptimizer optimizer; in TEST_F()
343 ArithmeticOptimizer optimizer; in TEST_F()
368 ArithmeticOptimizer optimizer; in TEST_F()
412 ArithmeticOptimizer optimizer; in TEST_F()
[all …]
Dmeta_optimizer.cc132 MK_OPT("arithmetic", new ArithmeticOptimizer(cfg_.arithmetic_optimization())); in MakeNewOptimizer()
189 MakeUnique<ArithmeticOptimizer>(cfg_.arithmetic_optimization())); in InitializeOptimizers()
Darithmetic_optimizer.cc3405 bool ArithmeticOptimizer::CanDedup(const NodeDef& node) const { in CanDedup()
3422 void ArithmeticOptimizer::DedupComputations() { in DedupComputations()
3512 void ArithmeticOptimizer::ForwardControlDependencies( in ForwardControlDependencies()
3527 Status ArithmeticOptimizer::SimplifyArithmeticOps(bool can_use_shapes) { in SimplifyArithmeticOps()
3649 Status ArithmeticOptimizer::Optimize(Cluster* /*cluster*/, in Optimize()
3691 void ArithmeticOptimizer::Feedback(Cluster* /*cluster*/, in Feedback()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmeta_optimizer.cc110 enabled_optimizers_["arithmetic"] = MakeUnique<ArithmeticOptimizer>(); in Init()