Searched refs:HloModuleMetadata (Results 1 – 6 of 6) sorted by relevance
41 TEST(HloModuleMetadata, RecordsPassStart) { in TEST() argument43 HloModuleMetadata module_metadata(&env); in TEST()51 TEST(HloModuleMetadata, RecordsPassEnd) { in TEST() argument53 HloModuleMetadata module_metadata(&env); in TEST()62 TEST(HloModuleMetadata, RecordsPassEndInNestedMetadata) { in TEST() argument64 HloModuleMetadata module_metadata(&env); in TEST()80 TEST(HloModuleMetadata, RecordPassEndReturnsNotFound) { in TEST() argument81 HloModuleMetadata module_metadata(tensorflow::Env::Default()); in TEST()91 TEST(HloModuleMetadata, SetsHloPassMetadataFields) { in TEST() argument92 HloModuleMetadata module_metadata(tensorflow::Env::Default()); in TEST()[all …]
25 StatusOr<HloPassMetadata*> HloModuleMetadata::GetCurrentHloPassMetadata() { in GetCurrentHloPassMetadata()36 Status HloModuleMetadata::MutateCurrentHloPassMetadata( in MutateCurrentHloPassMetadata()44 void HloModuleMetadata::RecordPassStart() { in RecordPassStart()51 Status HloModuleMetadata::RecordPassEnd() { in RecordPassEnd()59 void HloModuleMetadata::set_prepartitioning_metadata( in set_prepartitioning_metadata()60 const HloModuleMetadata& prepartitioning_metadata) { in set_prepartitioning_metadata()
32 class HloModuleMetadata {34 explicit HloModuleMetadata(tensorflow::Env* env) : env_(env) {} in HloModuleMetadata() function51 const HloModuleMetadata& prepartitioning_metadata);
372 const HloModuleMetadata& metadata() const { return metadata_; } in metadata()373 HloModuleMetadata* metadata() { return &metadata_; } in metadata()434 HloModuleMetadata metadata_;
536 // Uniquely identifies an HloModuleMetadata. Equal to the first unique_id539 // HloModuleMetadata with a different canonical_module_id.583 // the canonical_module_id of the HloModuleMetadata that this HloPassMetadata
267 // Dump HloModuleMetadata as a text proto for each HLO module.