Searched refs:HloVerifier (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 222 class HloVerifier : public HloModulePass {
224 explicit HloVerifier(
238 explicit HloVerifier(std::unique_ptr<TargetVerifierMetadata> target_metadata) in HloVerifier() function
241 ~HloVerifier() override = default;
|
| D | hlo_proto_util.cc | 46 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false) in CreateModuleFromProto()
|
| D | bfloat16_normalization_test.cc | 84 HloVerifier verifier(/*layout_sensitive=*/false, in Normalize()
|
| D | hlo_verifier.cc | 1468 StatusOr<bool> HloVerifier::Run(HloModule* module) { in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 184 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/tests/ |
| D | hlo_test_base.h | 66 HloVerifier verifier_;
304 HloVerifier& verifier() const { return *hlo_verifier_; } in verifier()
316 std::unique_ptr<HloVerifier> hlo_verifier_;
|
| D | token_hlo_test.cc | 83 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false) in XLA_TEST_F()
103 HloVerifier(/*layout_sensitive=*/false, /*allow_mixed_precision=*/false) in XLA_TEST_F()
|
| D | hlo_test_base.cc | 127 hlo_verifier_ = absl::make_unique<HloVerifier>( in HloTestBase()
|
| D | test_utils.cc | 569 return HloVerifier(/*layout_sensitive=*/layout_sensitive, in VerifyHloModule()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 251 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()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_extractor.cc | 153 HloVerifier verifier(/*layout_sensitive=*/false, in ExtractModule()
|