Searched refs:prepartitioning_metadata (Results 1 – 4 of 4) sorted by relevance
60 const HloModuleMetadata& prepartitioning_metadata) { in set_prepartitioning_metadata() argument62 prepartitioning_metadata.proto().canonical_module_id()); in set_prepartitioning_metadata()63 prepartitioning_metadata_ = prepartitioning_metadata.proto(); in set_prepartitioning_metadata()69 prepartitioning_metadata.running_passes_.begin(), in set_prepartitioning_metadata()70 prepartitioning_metadata.running_passes_.end()); in set_prepartitioning_metadata()72 prepartitioning_metadata.proto().pass_metadata()) { in set_prepartitioning_metadata()
46 const absl::optional<HloModuleMetadataProto>& prepartitioning_metadata() in prepartitioning_metadata() function51 const HloModuleMetadata& prepartitioning_metadata);
559 const absl::optional<HloModuleMetadataProto>& prepartitioning_metadata = in DumpHloModuleMetadataIfEnabled() local560 module->metadata().prepartitioning_metadata(); in DumpHloModuleMetadataIfEnabled()561 if (prepartitioning_metadata.has_value()) { in DumpHloModuleMetadataIfEnabled()562 DumpHloModuleMetadata(*prepartitioning_metadata, opts, in DumpHloModuleMetadataIfEnabled()
139 EXPECT_THAT(new_module_metadata.prepartitioning_metadata()->pass_metadata(), in TEST()