Home
last modified time | relevance | path

Searched refs:AddInvariantChecker (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_pass_pipeline_test.cc200 pipeline.AddInvariantChecker<BarBlowerUpper>(); in TEST_F()
210 pipeline.AddInvariantChecker<BarBlowerUpper>(); in TEST_F()
224 pipeline.AddInvariantChecker<BarBlowerUpper>(); in TEST_F()
Dhlo_pass_pipeline.h58 T& AddInvariantChecker(Args&&... args) { in AddInvariantChecker() function
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc184 pipeline.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in OptimizeHloModule()
212 pass.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in OptimizeHloModule()
271 pipeline.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in OptimizeHloModule()
308 pipeline.AddInvariantChecker<HloVerifier>( in OptimizeHloModule()
361 fusion.AddInvariantChecker<HloVerifier>( in OptimizeHloModule()
377 reduce_pipeline.AddInvariantChecker<HloVerifier>( in OptimizeHloModule()
407 pipeline.AddInvariantChecker<HloVerifier>( in PrepareHloModuleForIrEmitting()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc251 pipeline.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in RunHloPassesThroughLayoutAssn()
284 pass.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in RunHloPassesThroughLayoutAssn()
344 after_layout_assn.AddInvariantChecker<HloVerifier>( in RunHloPassesAfterLayoutAssn()
353 pass.AddInvariantChecker<HloVerifier>( in RunHloPassesAfterLayoutAssn()