Searched refs:module_metadata_ (Results 1 – 2 of 2) sorted by relevance
36 const HloModuleMetadataProto& proto() const { return module_metadata_; } in proto()55 module_metadata_.set_module_group_name(name); in set_module_group_name()58 module_metadata_.set_canonical_module_id(id); in set_canonical_module_id()61 module_metadata_.add_partitioned_module_ids(id); in add_partitioned_module_id()117 HloModuleMetadataProto module_metadata_; variable
45 HloPassMetadata* pass_metadata = module_metadata_.add_pass_metadata(); in RecordPassStart()61 module_metadata_.set_original_module_id( in set_prepartitioning_metadata()75 module_metadata_.add_pass_metadata(); in set_prepartitioning_metadata()