Searched refs:ArithmeticOptimizer (Results 1 – 8 of 8) 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 …]
34 class ArithmeticOptimizer : public GraphOptimizer {36 ArithmeticOptimizer() in ArithmeticOptimizer() function40 explicit ArithmeticOptimizer(RewriterConfig::Toggle opt_level) in ArithmeticOptimizer() function44 ~ArithmeticOptimizer() override {} in ~ArithmeticOptimizer()
105 ArithmeticOptimizer optimizer; in TEST_F()131 ArithmeticOptimizer optimizer; in TEST_F()174 ArithmeticOptimizer optimizer; in TEST_F()208 ArithmeticOptimizer optimizer; in TEST_F()239 ArithmeticOptimizer optimizer; in TEST_F()269 ArithmeticOptimizer optimizer; in TEST_F()304 ArithmeticOptimizer optimizer; in TEST_F()364 ArithmeticOptimizer optimizer; in TEST_F()409 ArithmeticOptimizer optimizer; in TEST_F()478 ArithmeticOptimizer optimizer; in TEST_F()[all …]
246 new ArithmeticOptimizer(cfg_.arithmetic_optimization())); in MakeNewOptimizer()412 MakeUnique<ArithmeticOptimizer>(cfg_.arithmetic_optimization())); in InitializeOptimizers()
4289 Status ArithmeticOptimizer::SimplifyArithmeticOps(bool can_use_shapes) { in SimplifyArithmeticOps()4420 Status ArithmeticOptimizer::Optimize(Cluster* /*cluster*/, in Optimize()
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"