Searched refs:ArithmeticOptimizer (Results 1 – 7 of 7) sorted by relevance
87 void DisableAddToAddNCombining(ArithmeticOptimizer* optimizer) { in DisableAddToAddNCombining()91 void EnableOnlyAddToAddNCombining(ArithmeticOptimizer* optimizer) { in EnableOnlyAddToAddNCombining()96 void EnableOnlyFoldConjugateIntoTranspose(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldConjugateIntoTranspose()101 void EnableOnlyFoldMultipleIntoConv(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldMultipleIntoConv()106 void EnableOnlyFoldTransposeIntoMatMul(ArithmeticOptimizer* optimizer) { in EnableOnlyFoldTransposeIntoMatMul()111 void EnableOnlyHoistCommonFactor(ArithmeticOptimizer* optimizer) { in EnableOnlyHoistCommonFactor()116 void EnableOnlyMinimizeBroadcasts(ArithmeticOptimizer* optimizer) { in EnableOnlyMinimizeBroadcasts()121 void EnableOnlyRemoveIdentityTranspose(ArithmeticOptimizer* optimizer) { in EnableOnlyRemoveIdentityTranspose()126 void EnableOnlyRemoveInvolution(ArithmeticOptimizer* optimizer) { in EnableOnlyRemoveInvolution()131 void EnableOnlyRemoveRedundantBitcast(ArithmeticOptimizer* optimizer) { in EnableOnlyRemoveRedundantBitcast()[all …]
33 class ArithmeticOptimizer : public GraphOptimizer {35 ArithmeticOptimizer() in ArithmeticOptimizer() function39 explicit ArithmeticOptimizer(RewriterConfig::Toggle opt_level) in ArithmeticOptimizer() function43 ~ArithmeticOptimizer() override {} in ~ArithmeticOptimizer()
102 ArithmeticOptimizer optimizer; in TEST_F()125 ArithmeticOptimizer optimizer; in TEST_F()170 ArithmeticOptimizer optimizer; in TEST_F()204 ArithmeticOptimizer optimizer; in TEST_F()252 ArithmeticOptimizer optimizer; in TEST_F()277 ArithmeticOptimizer optimizer; in TEST_F()324 ArithmeticOptimizer optimizer; in TEST_F()381 ArithmeticOptimizer optimizer; in TEST_F()463 ArithmeticOptimizer optimizer; in TEST_F()546 ArithmeticOptimizer optimizer; in TEST_F()[all …]
206 MK_OPT("arithmetic", new ArithmeticOptimizer(cfg_.arithmetic_optimization())); in MakeNewOptimizer()276 MakeUnique<ArithmeticOptimizer>(cfg_.arithmetic_optimization())); in InitializeOptimizers()
3664 Status ArithmeticOptimizer::SimplifyArithmeticOps(bool can_use_shapes) { in SimplifyArithmeticOps()3787 Status ArithmeticOptimizer::Optimize(Cluster* /*cluster*/, in Optimize()3829 void ArithmeticOptimizer::Feedback(Cluster* /*cluster*/, in Feedback()
48249 …name: "training/LossScaleOptimizer/gradients/tf_op_layer_Cast/Cast_grad/ArithmeticOptimizer/Reorde…48312 …input: "training/LossScaleOptimizer/gradients/tf_op_layer_Cast/Cast_grad/ArithmeticOptimizer/Reord…48337 …input: "training/LossScaleOptimizer/gradients/tf_op_layer_Cast/Cast_grad/ArithmeticOptimizer/Reord…48356 …input: "training/LossScaleOptimizer/gradients/tf_op_layer_Cast/Cast_grad/ArithmeticOptimizer/Reord…48378 name: "loss_1/ArithmeticOptimizer/AddOpsRewrite_add"48450 key: "_grappler:ArithmeticOptimizer:AddOpsRewriteStage"67338 input: "loss_1/ArithmeticOptimizer/AddOpsRewrite_add"
29752 name: "loss_1/ArithmeticOptimizer/HoistCommonFactor_Add_AddN"29878 name: "loss_1/ArithmeticOptimizer/HoistCommonFactor_Mul_AddN"29881 input: "loss_1/ArithmeticOptimizer/HoistCommonFactor_Add_AddN"44382 input: "loss_1/ArithmeticOptimizer/HoistCommonFactor_Mul_AddN"