Searched refs:HloVerifier (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | conditional_simplifier_test.cc | 209 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
268 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
314 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
358 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
404 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
470 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
511 HloVerifier v(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false); in TEST_F()
|
| D | while_loop_all_reduce_code_motion_test.cc | 90 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
189 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
268 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
384 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
476 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
554 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
634 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/true) in TEST_F()
|
| D | hlo_verifier.h | 263 class HloVerifier : public HloModulePass {
265 explicit HloVerifier(
279 explicit HloVerifier(std::unique_ptr<TargetVerifierMetadata> target_metadata) in HloVerifier() function
282 ~HloVerifier() override = default;
|
| D | hlo_proto_util.cc | 47 HloVerifier(/*layout_sensitive=*/false, in CreateModuleFromProto()
|
| D | memory_space_propagation_test.cc | 35 HloVerifier verifier_;
|
| D | bfloat16_normalization_test.cc | 86 HloVerifier verifier(/*layout_sensitive=*/false, in Normalize()
|
| D | bfloat16_propagation_test.cc | 1255 EXPECT_TRUE(HloVerifier(/*layout_sensitive=*/false, in TEST_F()
|
| D | hlo_verifier.cc | 1930 StatusOr<bool> HloVerifier::Run(HloModule* module) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | verified_hlo_module.h | 60 HloVerifier verifier_;
|
| D | token_hlo_test.cc | 82 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false) in XLA_TEST_F()
102 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false) in XLA_TEST_F()
|
| D | hlo_test_base.h | 307 HloVerifier& verifier() const { return *hlo_verifier_; } in verifier()
319 std::unique_ptr<HloVerifier> hlo_verifier_;
|
| D | hlo_test_base.cc | 108 hlo_verifier_ = absl::make_unique<HloVerifier>( in HloTestBase()
|
| D | test_utils.cc | 708 return HloVerifier(/*layout_sensitive=*/layout_sensitive, in VerifyHloModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | amdgpu_compiler.cc | 79 pipeline.AddInvariantCheckerDebug<HloVerifier>( in OptimizeHloConvolutionCanonicalization()
|
| D | nvptx_compiler.cc | 113 pipeline.AddInvariantCheckerDebug<HloVerifier>( in OptimizeHloConvolutionCanonicalization()
132 pass.AddInvariantCheckerDebug<HloVerifier>(/*layout_sensitive=*/false, in OptimizeHloConvolutionCanonicalization()
|
| D | gpu_compiler.cc | 154 pipeline.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in OptimizeHloModule()
221 pass.AddInvariantCheckerDebug<HloVerifier>( in OptimizeHloModule()
316 fusion.AddInvariantCheckerDebug<HloVerifier>( in OptimizeHloModule()
368 pipeline.AddInvariantCheckerDebug<HloVerifier>( in PrepareHloModuleForIrEmitting()
408 pipeline.AddInvariantCheckerDebug<HloVerifier>( in OptimizeHloPostLayoutAssignment()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 154 HloVerifier verifier(/*layout_sensitive=*/false, in ExtractModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 287 pipeline.AddInvariantChecker<HloVerifier>(/*layout_sensitive=*/false, in RunHloPassesThroughLayoutAssn()
346 pass.AddInvariantCheckerDebug<HloVerifier>(/*layout_sensitive=*/false, in RunHloPassesThroughLayoutAssn()
408 .AddInvariantCheckerDebug<HloVerifier>( in RunHloPassesAfterLayoutAssn()
417 pass.AddInvariantCheckerDebug<HloVerifier>( in RunHloPassesAfterLayoutAssn()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 58 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
62 pass.AddPass<HloVerifier>(/*layout_sensitive=*/false, in PartitionComputation()
|