Searched refs:merged_metadata (Results 1 – 3 of 3) sorted by relevance
176 std::vector<OpMetadata> merged_metadata(std::move(to_merge->metadata())); in MergeSharding() local177 merged_metadata.reserve(merged_metadata.size() + old.metadata().size()); in MergeSharding()180 metadata_set(merged_metadata.begin(), merged_metadata.end()); in MergeSharding()181 absl::c_copy_if(old.metadata(), std::back_inserter(merged_metadata), in MergeSharding()188 *to_merge = HloSharding::Tile(new_tile, merged_metadata); in MergeSharding()190 *to_merge = HloSharding::PartialTile(new_tile, merged_metadata); in MergeSharding()
714 std::unique_ptr<WritableFile> merged_metadata; in MergeBundles() local716 env->NewWritableFile(MetaFilename(merged_prefix), &merged_metadata)); in MergeBundles()718 table::TableBuilder builder(TableBuilderOptions(), merged_metadata.get()); in MergeBundles()731 status.Update(merged_metadata->Close()); in MergeBundles()
2607 std::map<int, llvm::MDNode*> merged_metadata = in EmitTransferElements() local2610 for (const auto& kind_md_pair : merged_metadata) { in EmitTransferElements()