Searched refs:saving_specs (Results 1 – 2 of 2) sorted by relevance
103 saving_specs; in BuildSavingSpec() local115 saving_specs[0][index].push_back(""); in BuildSavingSpec()124 saving_specs[saving_device_id][index].push_back(slice); in BuildSavingSpec()130 return saving_specs; in BuildSavingSpec()
182 saving_specs) { in ConditionalSave()230 const auto& it = saving_specs.find(device_id); in ConditionalSave()231 if (it == saving_specs.end()) { in ConditionalSave()418 TF_ASSIGN_OR_RETURN(auto saving_specs, BuildSavingSpec(metadata)); in ExpandSaveV2Op()423 ConditionalSave(save_v2, mesh, saving_specs)); in ExpandSaveV2Op()