Home
last modified time | relevance | path

Searched refs:AlgebraicSimplifier (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc72 AlgebraicSimplifier simplifier(default_options_); in TEST_F()
91 AlgebraicSimplifier simplifier(default_options_); in TEST_F()
113 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
131 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
151 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
172 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
195 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
217 ASSERT_TRUE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
239 EXPECT_FALSE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
257 EXPECT_FALSE(AlgebraicSimplifier(default_options_).Run(m.get()).ValueOrDie()); in TEST_F()
[all …]
Dalgebraic_simplifier.h188 class AlgebraicSimplifier : public HloModulePass {
192 explicit AlgebraicSimplifier(const AlgebraicSimplifierOptions& options) in AlgebraicSimplifier() function
194 ~AlgebraicSimplifier() override = default;
Dstable_sort_expander_test.cc247 AlgebraicSimplifier simplifier(AlgebraicSimplifierOptions( in TEST_F()
284 AlgebraicSimplifier simplifier(AlgebraicSimplifierOptions( in TEST_F()
Dalgebraic_simplifier.cc276 AlgebraicSimplifier* simplifier) in AlgebraicSimplifierVisitor()
374 AlgebraicSimplifier* simplifier);
543 AlgebraicSimplifier* simplifier_ = nullptr;
556 AlgebraicSimplifier* simplifier) { in Run()
1373 AlgebraicSimplifier* simplifier) { in BuildTupleConstant()
1463 AlgebraicSimplifier* simplifier) { in TryDivideToShift()
3687 AlgebraicSimplifier* simplifier) { in TryRemainderToAnd()
5608 StatusOr<bool> AlgebraicSimplifier::Run(HloModule* module) { in Run()
Dlayout_assignment_test.cc334 EXPECT_TRUE(AlgebraicSimplifier(options).Run(m.get()).ValueOrDie()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc243 pass.AddPass<AlgebraicSimplifier>(options); in OptimizeHloModule()
351 pipeline.AddPass<AlgebraicSimplifier>(options); in OptimizeHloModule()
432 pipeline.AddPass<HloPassFix<AlgebraicSimplifier>>(options); in OptimizeHloPostLayoutAssignment()
Dnvptx_compiler.cc148 pass.AddPass<AlgebraicSimplifier>(options); in OptimizeHloConvolutionCanonicalization()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc352 pass.AddPass<AlgebraicSimplifier>(options); in RunHloPassesThroughLayoutAssn()
424 pass.AddPass<HloPassFix<AlgebraicSimplifier>>(options); in RunHloPassesAfterLayoutAssn()